Shymaa2611 commited on
Commit ·
b128f83
1
Parent(s): 6ac2ca3
update
Browse files- inference.py +2 -3
inference.py
CHANGED
|
@@ -6,11 +6,10 @@ import os
|
|
| 6 |
|
| 7 |
def load_tokenizer_model():
|
| 8 |
os.environ['GDOWN_NO_COOKIES'] = '1'
|
| 9 |
-
|
| 10 |
-
folder_url = "https://drive.google.com/drive/folders/1DDJ9t-HfMrf6OLYim5bVrP20QgyOZahc"
|
| 11 |
gdown.download_folder(folder_url, output="ChatbotCheckpoint", quiet=False)
|
| 12 |
model_name="ChatbotCheckpoint"
|
| 13 |
-
model = GPT2LMHeadModel.from_pretrained(model_name
|
| 14 |
model.eval()
|
| 15 |
tokenizer=GPT2Tokenizer.from_pretrained(model_name)
|
| 16 |
tokenizer.pad_token=tokenizer.eos_token
|
|
|
|
| 6 |
|
| 7 |
def load_tokenizer_model():
|
| 8 |
os.environ['GDOWN_NO_COOKIES'] = '1'
|
| 9 |
+
folder_url = "https://drive.google.com/drive/folders/1DDJ9t-HfMrf6OLYim5bVrP20QgyOZahc?usp=drive_link"
|
|
|
|
| 10 |
gdown.download_folder(folder_url, output="ChatbotCheckpoint", quiet=False)
|
| 11 |
model_name="ChatbotCheckpoint"
|
| 12 |
+
model = GPT2LMHeadModel.from_pretrained(model_name)
|
| 13 |
model.eval()
|
| 14 |
tokenizer=GPT2Tokenizer.from_pretrained(model_name)
|
| 15 |
tokenizer.pad_token=tokenizer.eos_token
|