Shymaa2611 commited on
Commit
bf1d2cb
·
1 Parent(s): 58ce18a
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -9,7 +9,7 @@ os.makedirs(cache_dir, exist_ok=True)
9
  def load_tokenizer_model():
10
  folder_url = "https://drive.google.com/drive/folders/1DDJ9t-HfMrf6OLYim5bVrP20QgyOZahc?usp=drive_link"
11
  local_folder="ChatbotCheckpoint"
12
- if not os.path.exists(local_folder):
13
  print("Downloading model and tokenizer...")
14
  gdown.download_folder(folder_url, output=local_folder, quiet=False)
15
  else:
 
9
  def load_tokenizer_model():
10
  folder_url = "https://drive.google.com/drive/folders/1DDJ9t-HfMrf6OLYim5bVrP20QgyOZahc?usp=drive_link"
11
  local_folder="ChatbotCheckpoint"
12
+ if not os.path.exists(local_folder) or not os.listdir(local_folder):
13
  print("Downloading model and tokenizer...")
14
  gdown.download_folder(folder_url, output=local_folder, quiet=False)
15
  else: