Update app.py

app.py

@@ -20,4 +20,12 @@ def generate_text(request: Request):
     headers = {"Authorization": f"Bearer {HF_TOKEN}"}
     payload = {"inputs": request.prompt}
     response = req.post(API_URL, headers=headers, json=payload)
-    return {"response": response.json()[0]["generated_text"]}
+    result = response.json()
+    
+    # Handle different response formats
+    if isinstance(result, list) and len(result) > 0:
+        return {"response": result[0].get("generated_text", str(result))}
+    elif isinstance(result, dict):
+        return {"response": result.get("generated_text", str(result))}
+    else:
+        return {"response": str(result)}