Update app.py
Browse files
app.py
CHANGED
|
@@ -125,7 +125,7 @@ general_models, general_cfg = checkpoint_utils.load_model_ensemble(
|
|
| 125 |
arg_overrides=overrides
|
| 126 |
)
|
| 127 |
|
| 128 |
-
move models to gpu
|
| 129 |
move2gpu(caption_models, caption_cfg)
|
| 130 |
move2gpu(refcoco_models, refcoco_cfg)
|
| 131 |
move2gpu(vqa_models, vqa_cfg)
|
|
|
|
| 125 |
arg_overrides=overrides
|
| 126 |
)
|
| 127 |
|
| 128 |
+
# move models to gpu
|
| 129 |
move2gpu(caption_models, caption_cfg)
|
| 130 |
move2gpu(refcoco_models, refcoco_cfg)
|
| 131 |
move2gpu(vqa_models, vqa_cfg)
|