Spaces:

DurgaDeepak
/

UVIS

Running on Zero

DurgaDeepak commited on 23 days ago

Commit

c5b802f

verified ·

1 Parent(s): 03b47ac

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -76,7 +76,7 @@ def preload_models():
     """
     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")

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