yxchng
commited on
Commit
•
b07ca48
1
Parent(s):
eb905fc
fix requirements
Browse files- app.py +4 -1
- checkpoints/.test.py.swp +0 -0
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -1,8 +1,11 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
|
|
|
|
|
|
|
|
|
|
| 3 |
image_path = './image001.png'
|
| 4 |
sentence = 'spoon on the dish'
|
| 5 |
-
weights = '
|
| 6 |
device = 'cpu'
|
| 7 |
|
| 8 |
# pre-process the input image
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
|
| 3 |
+
import os
|
| 4 |
+
|
| 5 |
+
os.system('wget https://huggingface.co/yxchng/elia_refcoco/resolve/main/model_best_refcoco_0508.pth')
|
| 6 |
image_path = './image001.png'
|
| 7 |
sentence = 'spoon on the dish'
|
| 8 |
+
weights = 'model_best_refcoco_0508.pth'
|
| 9 |
device = 'cpu'
|
| 10 |
|
| 11 |
# pre-process the input image
|
checkpoints/.test.py.swp
DELETED
|
Binary file (12.3 kB)
|
|
|
requirements.txt
CHANGED
|
@@ -11,5 +11,5 @@ scipy
|
|
| 11 |
scikit-image
|
| 12 |
pycocotools==2.0.2
|
| 13 |
#opencv-python==4.5.3.56
|
| 14 |
-
tokenizers==0.
|
| 15 |
h5py
|
|
|
|
| 11 |
scikit-image
|
| 12 |
pycocotools==2.0.2
|
| 13 |
#opencv-python==4.5.3.56
|
| 14 |
+
tokenizers==0.8.1rc1
|
| 15 |
h5py
|