Update app.py
Browse files
app.py
CHANGED
@@ -7,13 +7,14 @@ from datasets import load_dataset
|
|
7 |
|
8 |
processor = AutoProcessor.from_pretrained("microsoft/git-base-vqav2")
|
9 |
model = AutoModelForCausalLM.from_pretrained("microsoft/git-base-vqav2")
|
10 |
-
|
11 |
|
12 |
-
|
13 |
features:['image','answers','question'],
|
14 |
-
num_rows : 50
|
|
|
15 |
|
16 |
-
file_path = hf_hub_download(repo_id=
|
17 |
image = Image.open(file_path).convert("RGB")
|
18 |
|
19 |
pixel_values = processor(images=image, return_tensors="pt").pixel_values
|
|
|
7 |
|
8 |
processor = AutoProcessor.from_pretrained("microsoft/git-base-vqav2")
|
9 |
model = AutoModelForCausalLM.from_pretrained("microsoft/git-base-vqav2")
|
10 |
+
Dataset="Multimodal-Fatima/OK-VQA_train"
|
11 |
|
12 |
+
Dataset({
|
13 |
features:['image','answers','question'],
|
14 |
+
num_rows : 50
|
15 |
+
})
|
16 |
|
17 |
+
file_path = hf_hub_download(repo_id=Dataset, repo_type="dataset")
|
18 |
image = Image.open(file_path).convert("RGB")
|
19 |
|
20 |
pixel_values = processor(images=image, return_tensors="pt").pixel_values
|