Spaces:
Sleeping
Sleeping
Update the model name
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from transformers import AutoModelForCausalLM, GPT2Tokenizer
|
|
3 |
|
4 |
|
5 |
def generate_text(sentence, max_length=100):
|
6 |
-
model_path = "
|
7 |
model_name = "datificate/gpt2-small-spanish"
|
8 |
|
9 |
# Load the model and the tokenizer
|
|
|
3 |
|
4 |
|
5 |
def generate_text(sentence, max_length=100):
|
6 |
+
model_path = "franzemil/bolivianlm"
|
7 |
model_name = "datificate/gpt2-small-spanish"
|
8 |
|
9 |
# Load the model and the tokenizer
|