Jiayi-Pan commited on
Commit
1f4c39e
1 Parent(s): a85807a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ examples =[
12
  ['dog.jpg', 'A black dog stands on a <mask>, green fields behind him.'],
13
  ]
14
  device = "cuda" if torch.cuda.is_available() else "cpu"
15
- model, img_transform, tokenizer, post_processor, plot_results = torch.hub.load('Jiayi-Pan/RefCloze_Pub', 'flickr_base_model')
16
  # model, img_transform, tokenizer, post_processor, plot_results = torch.hub.load('.', 'flickr_base_model', source='local')
17
  model = model.to(device)
18
 
 
12
  ['dog.jpg', 'A black dog stands on a <mask>, green fields behind him.'],
13
  ]
14
  device = "cuda" if torch.cuda.is_available() else "cpu"
15
+ model, img_transform, tokenizer, post_processor, plot_results = torch.hub.load('Jiayi-Pan/RefCloze_Pub', 'flickr_base_model', force_reload=True)
16
  # model, img_transform, tokenizer, post_processor, plot_results = torch.hub.load('.', 'flickr_base_model', source='local')
17
  model = model.to(device)
18