Spaces:
Sleeping
Sleeping
add distill endpoint
Browse files
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/
|
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)
|