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 |
|