Update app.py

app.py

@@ -11,22 +11,22 @@ 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()
-# print("=" * 50)
-# print(f"CUDA available: {torch.cuda.is_available()}")
-# print(f"torch version: {torch.__version__}")
-# print(f"CUDA version: {torch.version.cuda}")
-# print(f"mmcv version: {mmcv.__version__}")
-# print("=" * 50)
+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()
+print("=" * 50)
+print(f"CUDA available: {torch.cuda.is_available()}")
+print(f"torch version: {torch.__version__}")
+print(f"CUDA version: {torch.version.cuda}")
+print(f"mmcv version: {mmcv.__version__}")
+print("=" * 50)
 
 # init
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")