Update app.py

app.py

@@ -5,7 +5,7 @@ pipeline = pipeline(task="text-generation", model="Preetham04/text-generation")
 
 def predict(input_img):
     predictions = pipeline(input_img)
-    return  {p["labels"] for p in predictions}
+    return  {p["title"] for p in predictions}
 
 gradio_app = gr.Interface(
     predict,