update

inference.py

@@ -8,6 +8,12 @@ os.makedirs(cache_dir, exist_ok=True)
 
 def load_tokenizer_model():
     folder_url = "https://drive.google.com/drive/folders/1DDJ9t-HfMrf6OLYim5bVrP20QgyOZahc?usp=drive_link"
+    local_folder="ChatbotCheckpoint"
+    if not os.path.exists(local_folder):
+        print("Downloading model and tokenizer...")
+        gdown.download_folder(folder_url, output=local_folder, quiet=False)
+    else:
+        print("Model and tokenizer already cached, skipping download.")
     gdown.download_folder(folder_url, output="ChatbotCheckpoint", quiet=False)
     model_name = "ChatbotCheckpoint"
     model = GPT2LMHeadModel.from_pretrained(model_name)