Update app.py

app.py

@@ -7,8 +7,8 @@ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float
 pipe.to("cpu")  # Use "cuda" if GPU is available
 
 def generate_image(prompt, pipe):
-    # Generate the image using the pipeline
-    image = pipe(prompt).images[0]
+    # Generate the image using the pipeline with proper args handling
+    image = pipe(prompt, guidance_scale=7.5, num_inference_steps=50).images[0]
     return image
 
 def chatbot(prompt):