Spaces:
Running
Running
acecalisto3
commited on
Commit
•
128eddd
1
Parent(s):
5436192
Update app.py
Browse files
app.py
CHANGED
@@ -98,9 +98,9 @@ def chat_interface_with_agent(input_text, agent_name):
|
|
98 |
if agent_prompt is None:
|
99 |
return f"Agent {agent_name} not found."
|
100 |
|
101 |
-
model_name =
|
102 |
try:
|
103 |
-
model =
|
104 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
105 |
generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
|
106 |
except EnvironmentError as e:
|
|
|
98 |
if agent_prompt is None:
|
99 |
return f"Agent {agent_name} not found."
|
100 |
|
101 |
+
model_name ="MaziyarPanahi/Codestral-22B-v0.1-GGUF"
|
102 |
try:
|
103 |
+
model = AutoModel.from_pretrained("MaziyarPanahi/Codestral-22B-v0.1-GGUF")
|
104 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
105 |
generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
|
106 |
except EnvironmentError as e:
|