Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import requests
+import json
 import io
 import random
 import os
@@ -39,7 +40,9 @@ def extend_prompt(input_text):
         response.raise_for_status()
 
         output = response.json()[0]
-        return output
+        
+        prompt = output.get("generated_text", "")
+        return prompt
         
     except requests.exceptions.RequestException as e:
         gr.Warning("Cannot extend prompt: Please try again.")