Upload gradio_app.py

gradio_app.py

@@ -707,8 +707,8 @@ if __name__ == '__main__':
     </div>
     """
 
-    demo = gr.Interface(fn=my_gpu_function, inputs=[], outputs="text")
-    demo.launch()
+    #demo = gr.Interface(fn=my_gpu_function, inputs=[], outputs="text")
+    #demo.launch()
 
     example_is = get_example_img_list()
     example_ts = get_example_txt_list()
@@ -768,15 +768,17 @@ if __name__ == '__main__':
 
     # https://discuss.huggingface.co/t/how-to-serve-an-html-file/33921/2
     # create a FastAPI app
-    app = FastAPI()
+    #app = FastAPI()
     # create a static directory to store the static files
     static_dir = Path(SAVE_DIR).absolute()
     static_dir.mkdir(parents=True, exist_ok=True)
-    app.mount("/static", StaticFiles(directory=static_dir, html=True), name="static")
+    #app.mount("/static", StaticFiles(directory=static_dir, html=True), name="static")
     shutil.copytree('./assets/env_maps', os.path.join(static_dir, 'env_maps'), dirs_exist_ok=True)
 
+
     if args.low_vram_mode:
         torch.cuda.empty_cache()
     demo = build_app()
-    app = gr.mount_gradio_app(app, demo, path="/")
-    uvicorn.run(app, host=args.host, port=args.port)
+    demo.launch()
+    #app = gr.mount_gradio_app(app, demo, path="/")
+    #uvicorn.run(app, host=args.host, port=args.port)