Update app.py

app.py

@@ -7,10 +7,11 @@ from datasets import load_dataset
 
 processor = AutoProcessor.from_pretrained("microsoft/git-base-vqav2")
 model = AutoModelForCausalLM.from_pretrained("microsoft/git-base-vqav2")
+Model="Multimodal-Fatima/OK-VQA_train"
 
-dataset = load_dataset("question","answers","image", split="train")
+dataset = load_dataset(Model, split=["question","answers","image"])
 
-file_path = hf_hub_download(repo_id="Multimodal-Fatima/OK-VQA_train", repo_type=dataset)
+file_path = hf_hub_download(repo_id=Model, repo_type=dataset)
 image = Image.open(file_path).convert("RGB")
 
 pixel_values = processor(images=image, return_tensors="pt").pixel_values