Update app.py

app.py

@@ -1,18 +1,9 @@
-import gradio as gr
-from diffusers import DiffusionPipeline
-import torch
+import os
+import subprocess
 
-# Load the model
-pipe = DiffusionPipeline.from_pretrained("prompthero/openjourney-v4", torch_dtype=torch.float32).to("cpu")
-
-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()
+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}")