Update app.py

app.py

@@ -8,7 +8,20 @@ from datasets import load_dataset
 processor = AutoProcessor.from_pretrained("microsoft/git-base-vqav2")
 model = AutoModelForCausalLM.from_pretrained("microsoft/git-base-vqav2")
 
-dataset=load_dataset("Multimodal-Fatima/OK-VQA_train", split=['image', 'answers','question'],num_rows: 50)
+dataset=load_dataset("Multimodal-Fatima/OK-VQA_train")
+Dataset({
+    features:['image', 'answers','question'],
+    num_rows: 50
+})
+
+{'image':Image(decode=True,id=None),
+'answers':Sequence(feature=Value(dtype='int64',id_None),length=-1,id=None),
+'question':Value(dtype='string',id=None)}
+
+Dataset({
+    features:['input_ids','attention_mask','pixel_values','pixel_mask','labels'],
+    num_rows:50
+})
 
 file_path = hf_hub_download(repo_id=Dataset, repo_type="dataset")
 image = Image.open(file_path).convert("RGB")