Spaces:
Sleeping
Sleeping
MinxuanQin
commited on
Commit
•
6f67cca
1
Parent(s):
1ec4aa4
fix bug in model loading
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ config = ViltConfig.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
|
|
11 |
processor = ViltProcessor.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
|
12 |
model = ViltForQuestionAnswering.from_pretrained("Minqin/carets_vqa_finetuned")
|
13 |
|
14 |
-
orig_model = ViltForQuestionAnswering("dandelin/vilt-b32-finetuned-vqa")
|
15 |
|
16 |
uploaded_file = st.file_uploader("Please upload one image (jpg)", type="jpg")
|
17 |
|
|
|
11 |
processor = ViltProcessor.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
|
12 |
model = ViltForQuestionAnswering.from_pretrained("Minqin/carets_vqa_finetuned")
|
13 |
|
14 |
+
orig_model = ViltForQuestionAnswering.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
|
15 |
|
16 |
uploaded_file = st.file_uploader("Please upload one image (jpg)", type="jpg")
|
17 |
|