Update app.py

app.py

@@ -19,7 +19,7 @@ def inference(prompt, negative_prompt, num_samples, height=512, width=512, num_i
         negative_prompt=negative_prompt,
         num_images_per_prompt=int(num_samples),
         num_inference_steps=int(num_inference_steps), guidance_scale=guidance_scale,
-        generator="cpu"  # Use the CPU for inference
+        generator=torch.device("cpu")  # Use the CPU for inference
     ).images
 
 with gr.Blocks() as demo: