Vitrous commited on
Commit
35fb342
·
verified ·
1 Parent(s): 717f03d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,8 +29,8 @@ def mistral_model():
29
  """
30
  if torch.cuda.is_available():
31
  model_id = "/code/mistral/"
32
- model = AutoModelForCausalLM.from_pretrained("/code/mistral/")
33
- tokenizer = AutoTokenizer.from_pretrained("/code/mistral/")
34
 
35
  return model,tokenizer
36
 
 
29
  """
30
  if torch.cuda.is_available():
31
  model_id = "/code/mistral/"
32
+ model = AutoModelForCausalLM.from_pretrained("/home/user/app/mistral/")
33
+ tokenizer = AutoTokenizer.from_pretrained("/home/user/app/mistral/")
34
 
35
  return model,tokenizer
36