Update app.py
Browse files
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 = '
|
| 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
|