start of space

Dockerfile

@@ -48,7 +48,7 @@ RUN pip install --no-cache-dir -U \
     opencv-python-headless==4.8.0.74 \
     Pillow==9.5.0 \
     colored==1.4.4
-RUN pip install --no-cache-dir -U gradio==3.37.0
+RUN pip install --no-cache-dir -U gradio==4.43.0
 
 COPY --chown=1000 . ${HOME}/app
 RUN git clone https://github.com/facebookresearch/CutLER.git

model.py

@@ -91,6 +91,7 @@ def load_model(score_threshold: float) -> VisualizationDemo:
     model_dir.mkdir(exist_ok=True)
     weight_path = model_dir / WEIGHT_URL.split('/')[-1]
     print("***weight_path***", weight_path)
+    print("torch.cuda.is_available()", torch.cuda.is_available())
 
     if not weight_path.exists():
         subprocess.run(shlex.split(f'wget {WEIGHT_URL} -O {weight_path}'))
@@ -122,4 +123,5 @@ def run_model(image_path: str, score_threshold: float = 0.5) -> np.ndarray:
     _, res = model.run_on_image(image)
     print("***END***")
     print("***res.get_image()***", res.get_image())
+    print("***type(res.get_image())***", type(res.get_image()))
     return res.get_image()