Update app.py

app.py

@@ -5,16 +5,18 @@ from diffusers import AutoPipelineForText2Image
 import gradio as gr
 import base64
 
-# Load the model once at the start
-print("Loading the Stable Diffusion model...")
-try:
-    model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
-    print("Model loaded successfully.")
-except Exception as e:
-    print(f"Error loading model: {e}")
-    model = None
+def load_model():
+    print("Loading the Stable Diffusion model...")
+    try:
+        model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
+        print("Model loaded successfully.")
+        return model
+    except Exception as e:
+        print(f"Error loading model: {e}")
+        return None
 
 def generate_image(prompt):
+    model = load_model()
     try:
         if model is None:
             raise ValueError("Model not loaded properly.")
@@ -74,9 +76,9 @@ gradio_interface = gr.Interface(
         gr.JSON(label="Character Dict"),
         gr.Dropdown(["oil painting", "sketch", "watercolor"], label="Selected Style")
     ],
-    outputs="json"
-)
+    outputs="json",
+    concurrency_limit=3)
 
 if __name__ == "__main__":
     print("Launching Gradio interface...")
-    gradio_interface.launch()
+    gradio_interface.launch()