acecalisto3 commited on
Commit
128eddd
1 Parent(s): 5436192

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = "Codestral-22B-v0.1-GGUF"
102
  try:
103
- model = AutoModelForCausalLM.from_pretrained(model_name)
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: