jw2yang commited on
Commit
3a3a697
1 Parent(s): 0f3ab7c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -48,11 +48,9 @@ if not os.path.exists(pretrained_pth_novg):
48
  build model
49
  '''
50
  model_last = BaseModel(opt, build_model(opt)).from_pretrained(pretrained_pth_last).eval().cuda()
51
- model_cap = BaseModel(opt, build_model(opt)).from_pretrained(pretrained_pth_novg).eval().cuda()
52
 
53
  with torch.no_grad():
54
  model_last.model.sem_seg_head.predictor.lang_encoder.get_text_embeddings(["background", "background"], is_eval=True)
55
- model_cap.model.sem_seg_head.predictor.lang_encoder.get_text_embeddings(["background", "background"], is_eval=True)
56
 
57
  '''
58
  inference model
 
48
  build model
49
  '''
50
  model_last = BaseModel(opt, build_model(opt)).from_pretrained(pretrained_pth_last).eval().cuda()
 
51
 
52
  with torch.no_grad():
53
  model_last.model.sem_seg_head.predictor.lang_encoder.get_text_embeddings(["background", "background"], is_eval=True)
 
54
 
55
  '''
56
  inference model