Spaces:
Running
Running
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)
|