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.")