Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ from airllm import AirLLMLlama2
|
|
25 |
|
26 |
MAX_LENGTH = 128
|
27 |
# could use hugging face model repo id:
|
28 |
-
model = AirLLMLlama2("
|
29 |
|
30 |
|
31 |
# tokenizer = AutoTokenizer.from_pretrained(model_id, token= token)
|
|
|
25 |
|
26 |
MAX_LENGTH = 128
|
27 |
# could use hugging face model repo id:
|
28 |
+
model = AirLLMLlama2("CohereForAI/c4ai-command-r-plus-4bit")
|
29 |
|
30 |
|
31 |
# tokenizer = AutoTokenizer.from_pretrained(model_id, token= token)
|