Update app.py

app.py

@@ -21,6 +21,8 @@ def install_cuda_toolkit():
     # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
     os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
 
+install_cuda_toolkit()
+
 
 def load_file(path):
 
@@ -56,8 +58,6 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
 
 
 if __name__ == "__main__":	
-	
-    install_cuda_toolkit()
     
     demo.launch(
         server_name='0.0.0.0',