Update app.py

app.py

@@ -14,10 +14,11 @@ class TextInput(BaseModel):
 class EmbeddingOutput(BaseModel):
     embedding: List[float]
 
+
+config = ModelConfig()
+model = ProdFeatureEncoder(config=config)
+
 @app.get("/encode_text/{text}", response_model=EmbeddingOutput)
-async def encode_text(text_input: TextInput):
-    config = ModelConfig()
-    model = ProdFeatureEncoder(config=config)
-    embedding = model(text_input.text)
+async def encode_text(text: str):
+    embedding = model(text)
     return {"embedding": embedding.tolist()}
-