Update app.py

app.py

@@ -27,8 +27,8 @@ pipe = pipeline("text-generation", model="Preetham04/text-generation")
 
 def generate(text):
     predictions = pipe(text)
-    return  {p[0]["title"] for p in predictions}
-    #return pipe(text)[0]["title"]
+    print(predictions)  # To see the structure of predictions
+    return {p["generated_text"] for p in predictions}
 
 
 with gr.Blocks() as demo: