remove saving files

app.py

@@ -31,14 +31,14 @@ def compose(prompt: str, filename: str = "generated_post"):
     
     # Generate the image
     image = image_gen.generate_image(prompt=richify_prompt(prompt))
-    # Save the image
-    image_path = save_image(image, filename=filename)
-    print(f"Image saved at: {image_path}")
+    # # Save the image
+    # image_path = save_image(image, filename=filename)
+    # print(f"Image saved at: {image_path}")
     caption = text_gen.generate_caption(prompt=prompt)
     # Save the caption
-    caption_path = save_caption(caption, filename=filename)
-    print(f"Caption saved at: {caption_path}")
-    return image_path, caption
+    # caption_path = save_caption(caption, filename=filename)
+    # print(f"Caption saved at: {caption_path}")
+    return image, caption
 
 
 if __name__ == '__main__':
@@ -47,7 +47,7 @@ if __name__ == '__main__':
         fn=compose,
         inputs=gr.Textbox(lines=5, label="Prompt", placeholder="Enter your prompt here..."),
         outputs=[
-            gr.Image(type="filepath", label="Generated Image"),
+            gr.Image(label="Generated Image"),
             gr.Textbox(label="Generated Caption")
         ],
         title="From Words to Reels",