Update app.py

app.py

@@ -5,6 +5,7 @@ import os
 import os.path as osp
 import time
 import gradio as gr
+os.environ['TORCH_CUDA_ARCH_LIST']="7.5;8.6"
 
 os.environ["GRADIO_TEMP_DIR"] = "./gradio_tmp"
 
@@ -114,4 +115,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(debug=DEBUG, share=True)
+    demo.launch()