Spaces:

anzorq
/

finetuned_diffusion

Runtime error

anzorq commited on Oct 26, 2022

Commit

cebef55

•

1 Parent(s): 7bd0a5a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ from diffusers import StableDiffusionPipeline
 import gradio as gr
 import torch
-device = "cuda:0" if torch.cuda.is_available() else "cpu"
 pipe = StableDiffusionPipeline.from_pretrained("nitrosocke/Arcane-Diffusion", torch_dtype=torch.float16)
 if torch.cuda.is_available():
@@ -45,7 +45,7 @@ with gr.Blocks() as demo:
             run = gr.Button(value="Run")
             gr.Markdown(f"Running on: {device}")
         with gr.Column():
-            gallery = gr.Gallery()
     run.click(inference, inputs=[prompt, guidance, steps], outputs=gallery)
     gr.Examples([

 import gradio as gr
 import torch
+device = "GPU 🔥" if torch.cuda.is_available() else "CPU 🥶"
 pipe = StableDiffusionPipeline.from_pretrained("nitrosocke/Arcane-Diffusion", torch_dtype=torch.float16)
 if torch.cuda.is_available():
             run = gr.Button(value="Run")
             gr.Markdown(f"Running on: {device}")
         with gr.Column():
+            gallery = gr.Gallery(height=512)
     run.click(inference, inputs=[prompt, guidance, steps], outputs=gallery)
     gr.Examples([