Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -10,7 +10,7 @@ import time
|
|
| 10 |
import psutil
|
| 11 |
import random
|
| 12 |
from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
| 13 |
-
|
| 14 |
|
| 15 |
start_time = time.time()
|
| 16 |
current_steps = 25
|
|
@@ -57,7 +57,7 @@ def error_str(error, title="Error"):
|
|
| 57 |
else ""
|
| 58 |
)
|
| 59 |
|
| 60 |
-
|
| 61 |
def inference(
|
| 62 |
prompt,
|
| 63 |
neg_prompt,
|
|
@@ -259,5 +259,5 @@ with gr.Blocks(css="style.css") as demo:
|
|
| 259 |
|
| 260 |
print(f"Space built in {time.time() - start_time:.2f} seconds")
|
| 261 |
|
| 262 |
-
demo.queue(concurrency_count=1)
|
| 263 |
demo.launch(show_api=False)
|
|
|
|
| 10 |
import psutil
|
| 11 |
import random
|
| 12 |
from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
| 13 |
+
import spaces
|
| 14 |
|
| 15 |
start_time = time.time()
|
| 16 |
current_steps = 25
|
|
|
|
| 57 |
else ""
|
| 58 |
)
|
| 59 |
|
| 60 |
+
@spaces.GPU
|
| 61 |
def inference(
|
| 62 |
prompt,
|
| 63 |
neg_prompt,
|
|
|
|
| 259 |
|
| 260 |
print(f"Space built in {time.time() - start_time:.2f} seconds")
|
| 261 |
|
| 262 |
+
demo.queue(concurrency_count=1, api_open=False)
|
| 263 |
demo.launch(show_api=False)
|