update

gradio_demo/inference.py → app.py

@@ -126,7 +126,7 @@ def process_image(
 
     return result_image[:, :, ::-1], json_file, fig_file
     
-
+@spaces.GPU()
 def run_demo():
     """create the Gradio demo interface"""
     css = """
@@ -248,5 +248,4 @@ def run_demo():
     
     return demo
 
-if __name__ == "__main__":
-    run_demo().launch()
+run_demo().launch()