Hugging Face's logo

Spaces:
SurajSingh
/
aiml_gr16
Runtime error

App Files Community
SurajSingh commited on
Commit
235cb56
·
verified ·
1 Parent(s): e27bd54

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -15,7 +15,7 @@ app.add_middleware(
15
  )
16
 
17
  def load_model(model_path):
18
- model = GPT2LMHeadModel.from_pretrained(model_path,cache_dir="./cache")
19
  return model
20
 
21
 
 
15
  )
16
 
17
  def load_model(model_path):
18
+ model = GPT2LMHeadModel.from_pretrained(model_path)
19
  return model
20
 
21