panda47 commited on
Commit
360cf9c
Β·
1 Parent(s): 1c2976c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -9,7 +9,9 @@ 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(Model, split=["question+answers+image[1%:50%]"])
 
 
13
 
14
  file_path = hf_hub_download(repo_id=Model, repo_type=dataset)
15
  image = Image.open(file_path).convert("RGB")
 
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")