Update app.py

app.py

@@ -27,7 +27,7 @@ def predict(prompt):
         top_k=50,                 # Limit sampling to top-K probabilities
         top_p=0.95,               # Limit sampling to top-P probabilities (nucleus sampling)
         truncation=True            # Truncate text if it exceeds max length
-    )
+    )[0]["generated_text"]
     
     return completions