yxchng commited on
Commit
34d4db7
1 Parent(s): a1eaffc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ print("C")
14
  os.system('wget https://huggingface.co/yxchng/magnet_weights/resolve/main/magnet_refcoco.pth')
15
  image_path = './image001.png'
16
  sentence = 'spoon on the dish'
17
- weights = 'model_best_refcoco_0508.pth'
18
  device = 'cpu'
19
 
20
  # pre-process the input image
 
14
  os.system('wget https://huggingface.co/yxchng/magnet_weights/resolve/main/magnet_refcoco.pth')
15
  image_path = './image001.png'
16
  sentence = 'spoon on the dish'
17
+ weights = 'magnet_refcoco.pth'
18
  device = 'cpu'
19
 
20
  # pre-process the input image