Update app.py

app.py

@@ -22,15 +22,13 @@ model_id = "CompVis/stable-diffusion-v1-4"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 context = autocast if device == "cuda" else nullcontext
 
-pipe = StableDiffusionPipeline.from_pretrained(model_id,torch_dtype=torch.float16,use_auth_token=token).to(device)
+pipe = StableDiffusionPipeline.from_pretrained(model_id,use_auth_token=token).to(device)
 
 
 def infer(prompt,samples):
-    all_images = [] 
     with context(device):
-        images = pipe(samples*[prompt],num_inference_steps=50, guidance_scale=7.5, height=400, width=400).sample
-        all_images.extend(images)
-    return all_images
+        images = pipe(samples*[prompt],num_inference_steps=50, guidance_scale=7.5, height=400, width=400).images
+    return images
 
 
 demo = gr.Blocks()