Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
24 |
-
|
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 =
|
152 |
-
model =
|
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 |
|