Update app.py

app.py

@@ -7,8 +7,14 @@ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float
 pipe.to("cpu")  # Use "cuda" if GPU is available
 
 def generate_image(prompt):
-    image = pipe(prompt).images[0]
+    # Ensure prompt is passed correctly
+    inputs = pipe.tokenizer(prompt, return_tensors="pt")
+    with torch.no_grad():
+        # Ensure inputs are provided correctly
+        output = pipe(prompt, **inputs)
+    image = output.images[0]
     return image
+
 def chatbot(prompt):
     # Generate the image based on the user's input
     image = generate_image(prompt)