Nan Xue commited on
Commit ·
6bd00c7
1
Parent(s): c3190ee
update
Browse files
app.py
CHANGED
|
@@ -31,6 +31,7 @@ def stop_run():
|
|
| 31 |
gr.update(visible=False),
|
| 32 |
)
|
| 33 |
|
|
|
|
| 34 |
def process_image(
|
| 35 |
input_image,
|
| 36 |
model_name='scalelsd-vitbase-v2-train-sa1b.pt',
|
|
@@ -131,7 +132,6 @@ def process_image(
|
|
| 131 |
|
| 132 |
return result_image[:, :, ::-1], json_file, fig_file
|
| 133 |
|
| 134 |
-
@spaces.GPU()
|
| 135 |
def run_demo():
|
| 136 |
"""create the Gradio demo interface"""
|
| 137 |
css = """
|
|
|
|
| 31 |
gr.update(visible=False),
|
| 32 |
)
|
| 33 |
|
| 34 |
+
@spaces.GPU()
|
| 35 |
def process_image(
|
| 36 |
input_image,
|
| 37 |
model_name='scalelsd-vitbase-v2-train-sa1b.pt',
|
|
|
|
| 132 |
|
| 133 |
return result_image[:, :, ::-1], json_file, fig_file
|
| 134 |
|
|
|
|
| 135 |
def run_demo():
|
| 136 |
"""create the Gradio demo interface"""
|
| 137 |
css = """
|