Update app.py

app.py

@@ -1,3 +1,14 @@
+def install_mmcv():
+    try:
+        subprocess.run([
+            "pip", "install", "mmcv-full==1.7.2",
+            "-f", "https://download.openmmlab.com/mmcv/dist/cu121/torch2.1.0/"
+        ], check=True)
+    except subprocess.CalledProcessError as e:
+        print("Failed to install mmcv-full:", e)
+
+install_mmcv()
+import mmcv
 import gradio as gr
 import numpy as np
 # import spaces
@@ -11,17 +22,7 @@ from net.dornet import Net
 from net.dornet_ddp import Net_ddp
 import subprocess
 
-def install_mmcv():
-    try:
-        subprocess.run([
-            "pip", "install", "mmcv-full==1.7.2",
-            "-f", "https://download.openmmlab.com/mmcv/dist/cu121/torch2.1.0/"
-        ], check=True)
-    except subprocess.CalledProcessError as e:
-        print("Failed to install mmcv-full:", e)
 
-install_mmcv()
-import mmcv
 
 print("=" * 50)
 print(f"CUDA available: {torch.cuda.is_available()}")