Merlintxu commited on
Commit
d49c9f2
·
1 Parent(s): 1470252

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -9,10 +9,12 @@ MODELS = {
9
  "T5": "lmsys/fastchat-t5-3b-v1.0",
10
  "LSpanishGPT2": "PlanTL-GOB-ES/gpt2-large-bne",
11
  "GPT2": "datificate/gpt2-small-spanish",
 
12
  }
13
 
14
  # Define your function
15
  def generate_and_analyze(model_name, input_text):
 
16
  tokenizer = AutoTokenizer.from_pretrained(model)
17
  TOKENIZERS[model_name] = tokenizer
18
 
 
9
  "T5": "lmsys/fastchat-t5-3b-v1.0",
10
  "LSpanishGPT2": "PlanTL-GOB-ES/gpt2-large-bne",
11
  "GPT2": "datificate/gpt2-small-spanish",
12
+ "OpenAssistant": "OpenAssistant"
13
  }
14
 
15
  # Define your function
16
  def generate_and_analyze(model_name, input_text):
17
+ model= MODELS[model_name]
18
  tokenizer = AutoTokenizer.from_pretrained(model)
19
  TOKENIZERS[model_name] = tokenizer
20