Spaces:
Runtime error
Runtime error
add m endpoint
Browse files
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/
|
15 |
-
distill_model_path = "hunthinn/
|
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)
|