Update app.py

app.py

@@ -3,21 +3,23 @@ import torch
 from diffusers import StableDiffusionPipeline
 
 # Load the model (replace with your actual model path)
-model_name = "stabilityai/stable-diffusion-2-1"  # Example model; replace with your own if needed
+model_name = "alvdansen/midsommarcartoon"  # Using the specified model
+device = "cpu"  # Ensure we are using CPU
 
-# Force the model to run on CPU
-device = "cpu"
 print(f"Using device: {device}")
 
 try:
     # Load the Stable Diffusion model
+    print("Loading model...")
     pipe = StableDiffusionPipeline.from_pretrained(model_name).to(device)
+    print("Model loaded successfully.")
 except Exception as e:
     print(f"Error loading model: {e}")
     raise
 
 def generate_image(prompt):
     try:
+        print(f"Generating image for prompt: {prompt}")
         # Generate image based on the prompt
         with torch.no_grad():
             image = pipe(prompt).images[0]
@@ -29,7 +31,7 @@ def generate_image(prompt):
 # Gradio interface setup
 gr.Interface(
     fn=generate_image,
-    inputs=gr.Textbox(label="Enter your prompt (e.g., 'doodle stickman running')"),
+    inputs=gr.Textbox(label="Enter your prompt (e.g., 'stickman in a field')"),
     outputs=gr.Image(type="pil"),
     title="Doodle Stickman Image Generator",
     description="Generate doodle-style stickman images from text prompts."