Amitontheweb commited on
Commit
2cc0ba7
·
verified ·
1 Parent(s): 2b7c140

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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_Mistral = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
24
- #model_Mistral = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
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