Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import torch
+import os
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 from typing import Optional
 from my_model.captioner.image_captioning import ImageCaptioningModel
@@ -20,6 +21,7 @@ class KBVQA():
         self.captioner = None
         self.detector = None
         self.kbvqa_model = None
+        self.access_token = os.getenv("HUGGINGFACE_TOKEN")
       #  self.kbvqa_model_loaded = self.all_models_loaded()