Commit
•
43bcc57
1
Parent(s):
84bf924
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
-
import
|
| 3 |
|
| 4 |
from pipeline_diffusehigh_sdxl import DiffuseHighSDXLPipeline
|
| 5 |
pipeline = DiffuseHighSDXLPipeline.from_pretrained(
|
|
@@ -7,7 +7,7 @@ pipeline = DiffuseHighSDXLPipeline.from_pretrained(
|
|
| 7 |
).to("cuda")
|
| 8 |
|
| 9 |
|
| 10 |
-
@
|
| 11 |
def process_(
|
| 12 |
prompt="",
|
| 13 |
target_height=[1536, 2048],
|
|
@@ -27,6 +27,7 @@ def process_(
|
|
| 27 |
).images[0]
|
| 28 |
|
| 29 |
return image
|
|
|
|
| 30 |
def create_demo():
|
| 31 |
with gr.Blocks(theme="bethecloud/storj_theme") as demo:
|
| 32 |
with gr.Row():
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
+
import spaces
|
| 3 |
|
| 4 |
from pipeline_diffusehigh_sdxl import DiffuseHighSDXLPipeline
|
| 5 |
pipeline = DiffuseHighSDXLPipeline.from_pretrained(
|
|
|
|
| 7 |
).to("cuda")
|
| 8 |
|
| 9 |
|
| 10 |
+
@spaces.GPU()
|
| 11 |
def process_(
|
| 12 |
prompt="",
|
| 13 |
target_height=[1536, 2048],
|
|
|
|
| 27 |
).images[0]
|
| 28 |
|
| 29 |
return image
|
| 30 |
+
|
| 31 |
def create_demo():
|
| 32 |
with gr.Blocks(theme="bethecloud/storj_theme") as demo:
|
| 33 |
with gr.Row():
|