Update app.py

app.py

@@ -1,9 +1,18 @@
-import os
-import subprocess
+import gradio as gr
+from diffusers import DiffusionPipeline
+import torch
 
-if __name__ == "__main__":
-    try:
-        # Run the Flask app using Gunicorn with 4 worker processes
-        subprocess.run(["gunicorn", "-w", "4", "-b", "0.0.0.0:7860", "myapp:myapp"], check=True)
-    except subprocess.CalledProcessError as e:
-        print(f"An error occurred while running the application: {e}")
+# Load the model
+pipe = DiffusionPipeline.from_pretrained("prompthero/openjourney-v4", torch_dtype=torch.float16).to("cuda")
+
+def generate_image(prompt):
+    image = pipe(prompt).images[0]
+    return image
+
+# Create Gradio interface
+gr.Interface(
+    fn=generate_image,
+    inputs=gr.Textbox(label="Enter your prompt"),
+    outputs="image",
+    title="Hugging Face Image Generator"
+).launch()