Hugging Face's logo

Spaces:
hunthinn
/
movie_title_gpt2
Runtime error

App Files Community
hunthinn commited on
Commit
3038e6a
•
1 Parent(s): d22d434

add distill endpoint

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ distill_model_path = "hunthinn/movie_title_gpt2"
17
 
18
  tokenizer = GPT2Tokenizer.from_pretrained(tokenizer_path)
19
  small_model = GPT2LMHeadModel.from_pretrained(small_model_path)
20
- # distill_model = GPT2LMHeadModel.from_pretrained(distill_model_path)
21
  medium_model = GPT2LMHeadModel.from_pretrained(medium_model_path)
22
 
23
  tokenizer.pad_token = tokenizer.eos_token
 
17
 
18
  tokenizer = GPT2Tokenizer.from_pretrained(tokenizer_path)
19
  small_model = GPT2LMHeadModel.from_pretrained(small_model_path)
20
+ distill_model = GPT2LMHeadModel.from_pretrained(distill_model_path)
21
  medium_model = GPT2LMHeadModel.from_pretrained(medium_model_path)
22
 
23
  tokenizer.pad_token = tokenizer.eos_token