Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -1,7 +1,6 @@
|
|
1 |
from diffusers import StableDiffusionPipeline
|
2 |
import gradio as gr
|
3 |
import torch
|
4 |
-
from utils import device
|
5 |
|
6 |
pipe = StableDiffusionPipeline.from_pretrained("nitrosocke/Arcane-Diffusion", torch_dtype=torch.float16)
|
7 |
if torch.cuda.is_available():
|
@@ -15,7 +14,7 @@ def inference(prompt, guidance, steps):
|
|
15 |
|
16 |
with gr.Blocks() as demo:
|
17 |
with gr.Row():
|
18 |
-
gr.Markdown(f"Running on: {device}")
|
19 |
with gr.Column():
|
20 |
prompt = gr.Textbox(label="prompt")
|
21 |
guidance = gr.Slider(label="guidance scale", value=7.5, maximum=15)
|
|
|
1 |
from diffusers import StableDiffusionPipeline
|
2 |
import gradio as gr
|
3 |
import torch
|
|
|
4 |
|
5 |
pipe = StableDiffusionPipeline.from_pretrained("nitrosocke/Arcane-Diffusion", torch_dtype=torch.float16)
|
6 |
if torch.cuda.is_available():
|
|
|
14 |
|
15 |
with gr.Blocks() as demo:
|
16 |
with gr.Row():
|
17 |
+
#gr.Markdown(f"Running on: {device}")
|
18 |
with gr.Column():
|
19 |
prompt = gr.Textbox(label="prompt")
|
20 |
guidance = gr.Slider(label="guidance scale", value=7.5, maximum=15)
|