Spaces:

CurHarsh
/

COC-VIZ

Sleeping

CurHarsh commited on May 24, 2023

Commit

ea5d667

1 Parent(s): 86d50e6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ def generate_coc_viz(model_name,
                  head,
                  alpha):
     # model_index = models.index(model_name)
-    checkpoint = ''
     args = PredictionArgs(
         model=model_name,
         image=image,

                  head,
                  alpha):
     # model_index = models.index(model_name)
+    checkpoint = 'coc_tiny_plain.pth.tar'
     args = PredictionArgs(
         model=model_name,
         image=image,