Spaces:

DurgaDeepak
/

UVIS

Running on Zero

DurgaDeepak commited on 28 days ago

Commit

dc71e71

verified ·

1 Parent(s): 330fb6c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -68,7 +68,7 @@ def preload_models():
     """
     from registry import get_model
     print("Warming up models for ZeroGPU...")
-    get_model("detection", "yolov5n-seg", device="cpu")
     get_model("segmentation", "deeplabv3_resnet50", device="cpu")
     get_model("depth", "midas_v21_small_256", device="cpu")

     """
     from registry import get_model
     print("Warming up models for ZeroGPU...")
+    get_model("detection", "yolov5n", device="cpu")
     get_model("segmentation", "deeplabv3_resnet50", device="cpu")
     get_model("depth", "midas_v21_small_256", device="cpu")