Shymaa2611 commited on
Commit
b128f83
·
1 Parent(s): 6ac2ca3
Files changed (1) hide show
  1. 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
- cache_dir="ChatbotCheckpoint"
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,cache_dir=cache_dir)
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