Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -34,7 +34,7 @@ class MistralToolCallingAgentTool:
|
|
34 |
description = "Uses Mistral-7B-Instruct to answer questions using code or reasoning"
|
35 |
|
36 |
def __init__(self):
|
37 |
-
self.model_id = "mistralai/Mistral-7B-Instruct"
|
38 |
self.tokenizer = AutoTokenizer.from_pretrained(self.model_id)
|
39 |
self.model = AutoModelForCausalLM.from_pretrained(self.model_id, device_map="auto", torch_dtype="auto")
|
40 |
self.pipeline = pipeline(
|
|
|
34 |
description = "Uses Mistral-7B-Instruct to answer questions using code or reasoning"
|
35 |
|
36 |
def __init__(self):
|
37 |
+
self.model_id = "mistralai/Mistral-7B-Instruct-v0.3"
|
38 |
self.tokenizer = AutoTokenizer.from_pretrained(self.model_id)
|
39 |
self.model = AutoModelForCausalLM.from_pretrained(self.model_id, device_map="auto", torch_dtype="auto")
|
40 |
self.pipeline = pipeline(
|