Upload app.py

app.py

@@ -13,7 +13,7 @@ class GenerationResponse(BaseModel):
 @app.post("/generate", response_model=GenerationResponse)
 def generate(req: GenerationRequest):
     out = f"hello world: {req}"
-    return out
+    return {"generated_text": out}
 
 @app.get("/health")
 def health():