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 |
|
|
|
20 |
tokenizer_gemma = AutoTokenizer.from_pretrained("google/gemma-2b")
|
21 |
model_gemma = AutoModelForCausalLM.from_pretrained("google/gemma-2b", token=token)
|
22 |
|
23 |
+
tokenizer_Mistral = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")
|
24 |
+
model_Mistral = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B")
|
25 |
|
26 |
# Define functions
|
27 |
|