Update app.py

app.py

@@ -20,5 +20,6 @@ model = ProdFeatureEncoder(config=config)
 
 @app.get("/encode_text/{text}", response_model=EmbeddingOutput)
 async def encode_text(text: str):
-    embedding = model(text)
+    with torch.no_grad():
+        embedding = model(text)
     return {"embedding": embedding.tolist()}