ikraamkb commited on
Commit
8b98a2c
·
verified ·
1 Parent(s): 461e409

ready for images

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -13,6 +13,8 @@ import numpy as np
13
 
14
  app = FastAPI()
15
 
 
 
16
  # Load VQA Model
17
  vqa_processor = ViltProcessor.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
18
  vqa_model = ViltForQuestionAnswering.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
 
13
 
14
  app = FastAPI()
15
 
16
+
17
+
18
  # Load VQA Model
19
  vqa_processor = ViltProcessor.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
20
  vqa_model = ViltForQuestionAnswering.from_pretrained("dandelin/vilt-b32-finetuned-vqa")