Hugging Face's logo

Spaces:
SurajSingh
/
aiml_gr16
Runtime error

App Files Community
SurajSingh commited on
Commit
00f2684
·
verified ·
1 Parent(s): 8f439a4

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -25,7 +25,7 @@ def load_tokenizer(checkpoint):
25
 
26
  model_path = r'./checkpoint/'
27
  model = load_model(model_path)
28
- tokenizer = load_tokenizer('gpt2', cache_dir="./cache")
29
 
30
 
31
  def generate_text(sequence, max_new_tokens):
 
25
 
26
  model_path = r'./checkpoint/'
27
  model = load_model(model_path)
28
+ tokenizer = load_tokenizer('gpt2')
29
 
30
 
31
  def generate_text(sequence, max_new_tokens):