Update app.py

app.py

@@ -9,8 +9,16 @@ from PIL import Image
 import torchvision.transforms as transforms
 from net.dornet import Net
 from net.dornet_ddp import Net_ddp
+import subprocess
 
+def install_mmcv():
+    url = "https://download.openmmlab.com/mmcv/dist/cu121/torch2.1/mmcv_full-1.7.2-cp310-cp310-manylinux1_x86_64.whl"
+    try:
+        subprocess.run(["pip", "install", url], check=True)
+    except subprocess.CalledProcessError as e:
+        print("Failed to install mmcv-full:", e)
 
+install_mmcv()
 print("=" * 50)
 print(f"CUDA available: {torch.cuda.is_available()}")
 print(f"torch version: {torch.__version__}")