Add model
Browse files
app.py
CHANGED
|
@@ -27,8 +27,8 @@ use_cuda = torch.cuda.is_available()
|
|
| 27 |
# use fp16 only when GPU is available
|
| 28 |
use_fp16 = False
|
| 29 |
|
| 30 |
-
|
| 31 |
-
|
| 32 |
|
| 33 |
# specify some options for evaluation
|
| 34 |
parser = options.get_generation_parser()
|
|
|
|
| 27 |
# use fp16 only when GPU is available
|
| 28 |
use_fp16 = False
|
| 29 |
|
| 30 |
+
os.system('wget https://huggingface.co/komleva/VQAmodel/resolve/main/checkpoint.best_score_0.9840.pt; '
|
| 31 |
+
'mkdir -p checkpoints; mv checkpoint.best_score_0.9840.pt checkpoints/checkpoint.best_score_0.9840.pt')
|
| 32 |
|
| 33 |
# specify some options for evaluation
|
| 34 |
parser = options.get_generation_parser()
|