Shymaa2611 commited on
Commit ·
bf1d2cb
1
Parent(s): 58ce18a
update
Browse files- 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:
|