Amitontheweb commited on
Commit
fe6f327
·
verified ·
1 Parent(s): 4792edb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -20,8 +20,8 @@ model_gpt2 = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
20
  tokenizer_gemma = AutoTokenizer.from_pretrained("google/gemma-2b")
21
  model_gemma = AutoModelForCausalLM.from_pretrained("google/gemma-2b", token=token)
22
 
23
- tokenizer_gpt-neo = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")
24
- model_gpt-neo = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B")
25
 
26
  # Define functions
27
 
@@ -148,8 +148,8 @@ def load_model(model_selected):
148
  model = model_gemma
149
 
150
  if model_selected == "Eleuther GPT Neo 1.3B":
151
- tokenizer = tokenizer_gpt-neo
152
- model = model_gpt-neo
153
 
154
 
155
 
 
20
  tokenizer_gemma = AutoTokenizer.from_pretrained("google/gemma-2b")
21
  model_gemma = AutoModelForCausalLM.from_pretrained("google/gemma-2b", token=token)
22
 
23
+ tokenizer_gptneo = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")
24
+ model_gptneo = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B")
25
 
26
  # Define functions
27
 
 
148
  model = model_gemma
149
 
150
  if model_selected == "Eleuther GPT Neo 1.3B":
151
+ tokenizer = tokenizer_gptneo
152
+ model = model_gptneo
153
 
154
 
155