franzemil commited on
Commit
287519a
·
1 Parent(s): 2ee4594

Update the model name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "models/franzemil/bolivianlm"
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