Update app.py

app.py

@@ -15,9 +15,9 @@ def home():
 
 @app.post("/generate")
 def generate_text(request: Request):
-    response = client.text_generation(
-        request.prompt,
+    response = client.chat_completion(
+        messages=[{"role": "user", "content": request.prompt}],
         model="HuggingFaceH4/zephyr-7b-beta",
-        max_new_tokens=256
+        max_tokens=256
     )
-    return {"response": response}
+    return {"response": response.choices[0].message.content}