hunthinn commited on
Commit
f3d38c9
1 Parent(s): 3038e6a

add distill endpoint

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ HfFolder.save_token(access_token)
13
  tokenizer_path = "gpt2"
14
  small_model_path = "hunthinn/movie_title_gpt2_small"
15
  medium_model_path = "hunthinn/movie_title_gpt2_medium"
16
- distill_model_path = "hunthinn/movie_title_gpt2"
17
 
18
  tokenizer = GPT2Tokenizer.from_pretrained(tokenizer_path)
19
  small_model = GPT2LMHeadModel.from_pretrained(small_model_path)
 
13
  tokenizer_path = "gpt2"
14
  small_model_path = "hunthinn/movie_title_gpt2_small"
15
  medium_model_path = "hunthinn/movie_title_gpt2_medium"
16
+ distill_model_path = "hunthinn/movie_title_gpt2_distill"
17
 
18
  tokenizer = GPT2Tokenizer.from_pretrained(tokenizer_path)
19
  small_model = GPT2LMHeadModel.from_pretrained(small_model_path)