SD pipeline

app.py

@@ -25,13 +25,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,use_auth_token=token).to(device)
+pipe = StableDiffusionPipeline.from_pretrained(model_id,use_auth_token=token).to(device)
 
-#
-# def infer_original(prompt,samples):
-#     with context(device):
-#         images = pipe(samples*[prompt], guidance_scale=7.5).images
-#     return images
+
+def infer_original(prompt,samples):
+    with context(device):
+        images = pipe(samples*[prompt], guidance_scale=7.5).images
+    return images
 
 
 
@@ -145,8 +145,8 @@ with demo:
 
                     gallery = gr.Gallery(label="Generated images", show_label=True).style(grid=(1, 3), height="auto")
                     # image_options = gr.Radio(label="Pick", interactive=True, choices=None, type="value")
-                    text.submit(infer, inputs=[text, samples], outputs=gallery)
-                    btn.click(infer, inputs=[text, samples], outputs=gallery, show_progress=True, status_tracker=None)
+                    text.submit(infer_original, inputs=[text, samples], outputs=gallery)
+                    btn.click(infer_original, inputs=[text, samples], outputs=gallery, show_progress=True, status_tracker=None)
 
 
     # Second Row - Backgrounds