Spaces:
Running
Running
acecalisto3
commited on
Commit
•
79ecef6
1
Parent(s):
128eddd
Update app.py
Browse files
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)
|