Spaces:
Running
on
Zero
Running
on
Zero
chore: use spaces.GPU for execute_image
Browse files- app.py +7 -2
- src/config/argument_config.py +3 -3
app.py
CHANGED
@@ -151,7 +151,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
|
151 |
output_image_paste_back.render()
|
152 |
# binding functions for buttons
|
153 |
process_button_retargeting.click(
|
154 |
-
fn=gradio_pipeline.execute_image,
|
|
|
155 |
inputs=[eye_retargeting_slider, lip_retargeting_slider],
|
156 |
outputs=[output_image, output_image_paste_back],
|
157 |
show_progress=True
|
@@ -174,4 +175,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
|
174 |
outputs=[eye_retargeting_slider, lip_retargeting_slider, retargeting_input_image]
|
175 |
)
|
176 |
|
177 |
-
demo.launch(
|
|
|
|
|
|
|
|
|
|
151 |
output_image_paste_back.render()
|
152 |
# binding functions for buttons
|
153 |
process_button_retargeting.click(
|
154 |
+
# fn=gradio_pipeline.execute_image,
|
155 |
+
fn=lambda *args: spaces.GPU()(gradio_pipeline.execute_image)(*args),
|
156 |
inputs=[eye_retargeting_slider, lip_retargeting_slider],
|
157 |
outputs=[output_image, output_image_paste_back],
|
158 |
show_progress=True
|
|
|
175 |
outputs=[eye_retargeting_slider, lip_retargeting_slider, retargeting_input_image]
|
176 |
)
|
177 |
|
178 |
+
demo.launch(
|
179 |
+
server_port=args.server_port,
|
180 |
+
share=args.share,
|
181 |
+
server_name=args.server_name
|
182 |
+
)
|
src/config/argument_config.py
CHANGED
@@ -39,6 +39,6 @@ class ArgumentConfig(PrintableConfig):
|
|
39 |
####################################
|
40 |
|
41 |
########## gradio arguments ##########
|
42 |
-
server_port: Annotated[int, tyro.conf.arg(aliases=["-p"])] =
|
43 |
-
share: bool =
|
44 |
-
server_name: str = "0.0.0.0"
|
|
|
39 |
####################################
|
40 |
|
41 |
########## gradio arguments ##########
|
42 |
+
server_port: Annotated[int, tyro.conf.arg(aliases=["-p"])] = 7860
|
43 |
+
share: bool = False
|
44 |
+
server_name: str = None # one can set "0.0.0.0" on local
|