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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -100,6 +100,7 @@ def chat_interface_with_agent(input_text, agent_name):
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)
 
100
 
101
  model_name ="MaziyarPanahi/Codestral-22B-v0.1-GGUF"
102
  try:
103
+ from transformers import AutoModel, AutoTokenizer # Import AutoModel here
104
  model = AutoModel.from_pretrained("MaziyarPanahi/Codestral-22B-v0.1-GGUF")
105
  tokenizer = AutoTokenizer.from_pretrained(model_name)
106
  generator = pipeline("text-generation", model=model, tokenizer=tokenizer)