Spaces:
Runtime error
Runtime error
robzchhangte
commited on
Commit
•
45bbd46
1
Parent(s):
03e032a
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
3 |
import gradio as gr
|
4 |
|
5 |
# Load pre-trained GPT-2 model and tokenizer
|
6 |
-
model_name = "robzchhangte/
|
7 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
8 |
# Set pad token to eos token
|
9 |
tokenizer.pad_token = tokenizer.eos_token
|
|
|
3 |
import gradio as gr
|
4 |
|
5 |
# Load pre-trained GPT-2 model and tokenizer
|
6 |
+
model_name = "robzchhangte/50k-MZGPT2-Vanilla"
|
7 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
8 |
# Set pad token to eos token
|
9 |
tokenizer.pad_token = tokenizer.eos_token
|