Update app.py

app.py

@@ -8,11 +8,10 @@ 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")
-train : Dataset({
+dataset=load_dataset("Multimodal-Fatima/OK-VQA_train", split='train')
+dataset({
     features: ['image', 'answers','question'],
-    num_rows: 8530
+    num_rows: 50
 })
 
 file_path = hf_hub_download(repo_id=Dataset, repo_type="dataset")