panda47 commited on
Commit
0c5ba4f
Β·
1 Parent(s): 94b1fad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
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
- Model="Multimodal-Fatima/OK-VQA_train"
11
 
12
- dataset = load_dataset({
13
  features:['image','answers','question'],
14
- num_rows : 50})
 
15
 
16
- file_path = hf_hub_download(repo_id=Model, repo_type=dataset)
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