CurHarsh commited on
Commit
3972689
·
1 Parent(s): 34fbab0

Initial commit

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,8 +19,8 @@ def generate_coc_viz(model_name,
19
  block,
20
  head,
21
  alpha):
22
- model_index = models.index(model_name)
23
- checkpoint = checkpoints[model_index]
24
  args = PredictionArgs(
25
  model=model_name,
26
  image=image,
 
19
  block,
20
  head,
21
  alpha):
22
+ # model_index = models.index(model_name)
23
+ checkpoint = ''
24
  args = PredictionArgs(
25
  model=model_name,
26
  image=image,