hunthinn commited on
Commit
e97e55a
1 Parent(s): a380c9e

add m endpoint

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,8 +11,8 @@ HfFolder.save_token(access_token)
11
 
12
  # Load the fine-tuned model and tokenizer
13
  tokenizer_path = "gpt2"
14
- small_model_path = "hunthinn/movie_title_gpt2"
15
- distill_model_path = "hunthinn/movie_title_gpt2"
16
  medium_model_path = "hunthinn/movie_title_gpt2"
17
 
18
  tokenizer = GPT2Tokenizer.from_pretrained(tokenizer_path)
 
11
 
12
  # Load the fine-tuned model and tokenizer
13
  tokenizer_path = "gpt2"
14
+ small_model_path = "hunthinn/movie_title_gpt2_small"
15
+ distill_model_path = "hunthinn/movie_title_gpt2_medium"
16
  medium_model_path = "hunthinn/movie_title_gpt2"
17
 
18
  tokenizer = GPT2Tokenizer.from_pretrained(tokenizer_path)