Update app.py

app.py

@@ -52,7 +52,7 @@ pipe.to(device)
 
 @spaces.GPU(duration=150)
 def generate_caption_and_image(image, f, p, d):
-    if f and p and d:
+    if f!=None and p!=None and d!=None:
         img = image.convert("RGB")
         # reader = easyocr.Reader(['en'])  
         # # result = reader.readtext(img)
@@ -91,7 +91,7 @@ def generate_caption_and_image(image, f, p, d):
         # Generate image based on the caption
         generated_image = pipe(prompt).images[0]
     
-        return prompt, generated_image
+        return generated_image
 
 # Gradio UI
 iface = gr.Interface(