Update app.py

app.py

@@ -26,7 +26,9 @@ pipe = pipeline("text-generation", model="Preetham04/text-generation")
 
 
 def generate(text):
-    return pipe(text)[0]["title"]
+    predictions = pipe(text)
+    return  {p[0]["title"] for p in predictions}
+    #return pipe(text)[0]["title"]
 
 
 with gr.Blocks() as demo: