Update app.py
Browse files
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("/
|
33 |
-
tokenizer = AutoTokenizer.from_pretrained("/
|
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 |
|