Vitrous commited on
Commit
8c6f39a
·
verified ·
1 Parent(s): 0977991

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -33,9 +33,8 @@ def mistral_model():
33
  else:
34
  print("CUDA is not available. CPU will be used.")
35
 
36
- model = AutoModelForCausalLM.from_pretrained(mistral_model)
37
-
38
- tokenizer = AutoTokenizer.from_pretrained(mistral_model)
39
 
40
  return model,tokenizer
41
 
 
33
  else:
34
  print("CUDA is not available. CPU will be used.")
35
 
36
+ tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
37
+ model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
 
38
 
39
  return model,tokenizer
40