Shymaa2611 commited on
Commit ·
3b2fb4f
1
Parent(s): 4eafc46
update
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -6,7 +6,7 @@ import gdown
|
|
| 6 |
def load_tokenizer_model():
|
| 7 |
cache_dir="ChatbotCheckpoint"
|
| 8 |
folder_url = "https://drive.google.com/drive/folders/1DDJ9t-HfMrf6OLYim5bVrP20QgyOZahc"
|
| 9 |
-
gdown.download_folder(folder_url, output="ChatbotCheckpoint")
|
| 10 |
model_name="ChatbotCheckpoint"
|
| 11 |
model = GPT2LMHeadModel.from_pretrained(model_name,cache_dir=cache_dir)
|
| 12 |
model.eval()
|
|
|
|
| 6 |
def load_tokenizer_model():
|
| 7 |
cache_dir="ChatbotCheckpoint"
|
| 8 |
folder_url = "https://drive.google.com/drive/folders/1DDJ9t-HfMrf6OLYim5bVrP20QgyOZahc"
|
| 9 |
+
gdown.download_folder(folder_url, output="ChatbotCheckpoint", quiet=False, no_cookies=True)
|
| 10 |
model_name="ChatbotCheckpoint"
|
| 11 |
model = GPT2LMHeadModel.from_pretrained(model_name,cache_dir=cache_dir)
|
| 12 |
model.eval()
|