Update app.py

app.py

@@ -1,18 +1,18 @@
 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()
+# 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
+import spaces
 import torch
 import os
 import cv2
@@ -40,7 +40,7 @@ model_ckpt_map = {
 }
 
 # load model
-# @spaces.GPU
+@spaces.GPU
 def load_model(model_type: str):
     global net
     ckpt_path = model_ckpt_map[model_type]
@@ -59,7 +59,7 @@ load_model("RGB-D-D")
 
 
 # data process
-# @spaces.GPU
+@spaces.GPU
 def preprocess_inputs(rgb_image: Image.Image, lr_depth: Image.Image):
     image = np.array(rgb_image.convert("RGB")).astype(np.float32)
     h, w, _ = image.shape
@@ -81,7 +81,7 @@ def preprocess_inputs(rgb_image: Image.Image, lr_depth: Image.Image):
 
 
 # model inference
-# @spaces.GPU
+@spaces.GPU
 @torch.no_grad()
 def infer(rgb_image: Image.Image, lr_depth: Image.Image, model_type: str):
     load_model(model_type)  # reset weight