Spaces:
Runtime error
Runtime error
Commit
·
267aac4
1
Parent(s):
0c5448c
update
Browse files- injection_main_HF.py +2 -2
injection_main_HF.py
CHANGED
@@ -516,7 +516,7 @@ if __name__ == "__main__":
|
|
516 |
if mode == "app":
|
517 |
# gradio
|
518 |
import gradio as gr
|
519 |
-
import spaces
|
520 |
|
521 |
# Load a pipeline
|
522 |
pipe = StableDiffusionPipeline.from_pretrained(
|
@@ -526,7 +526,7 @@ if __name__ == "__main__":
|
|
526 |
# Set up a DDIM scheduler
|
527 |
pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
|
528 |
|
529 |
-
@spaces.GPU()
|
530 |
def style_transfer_app(
|
531 |
prompt,
|
532 |
image,
|
|
|
516 |
if mode == "app":
|
517 |
# gradio
|
518 |
import gradio as gr
|
519 |
+
# import spaces
|
520 |
|
521 |
# Load a pipeline
|
522 |
pipe = StableDiffusionPipeline.from_pretrained(
|
|
|
526 |
# Set up a DDIM scheduler
|
527 |
pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
|
528 |
|
529 |
+
# @spaces.GPU()
|
530 |
def style_transfer_app(
|
531 |
prompt,
|
532 |
image,
|