Update app.py

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, request, jsonify
+from flask import Flask, request
 from flask_cors import CORS
 import os
 from huggingface_hub import InferenceClient
@@ -19,7 +19,16 @@ def generate_image(prompt, seed=1, model="prompthero/openjourney-v4"):
     try:
         # Generate the image using Hugging Face's inference API
         result_image = client.text_to_image(prompt=prompt, seed=seed, model=model)
-        return result_image
+
+        # Convert the result to a PIL Image
+        if isinstance(result_image, bytes):
+            # If the result is in bytes format
+            image = Image.open(BytesIO(result_image))
+        else:
+            # If the result is in another format, handle accordingly
+            raise ValueError("Received image in an unexpected format")
+
+        return image
     except Exception as e:
         print(f"Error generating image: {str(e)}")
         return None
@@ -47,7 +56,7 @@ gr.Interface(
     outputs="image",
     title="Image Generation with Hugging Face",
     description="Enter a prompt, seed, and model name to generate an image."
-).launch()  # Launch the Gradio interface
+).launch(share=True)  # Launch the Gradio interface
 
 # Add this block to make sure your app runs when called
 if __name__ == "__main__":