Update app.py

app.py

@@ -2,11 +2,11 @@ import gradio as gr
 from diffusers import StableDiffusionPipeline
 import torch
 
-
+# Make sure to specify the correct model
 pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4-original", torch_dtype=torch.float32)
-pipe.to("cpu")  # Make sure to use CPU
+pipe.to("cpu")  # Run on CPU (or change to "cuda" if you have a GPU)
 
-# Define the function for inference
+# Inference function
 def infer(prompt, guidance_scale=7.5, num_inference_steps=50):
     # Generate image from the prompt
     image = pipe(prompt, guidance_scale=guidance_scale, num_inference_steps=num_inference_steps).images[0]