Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -30,7 +30,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
| 30 |
# SMALL LLM (POST-EDITOR)
|
| 31 |
# ============================================================
|
| 32 |
|
| 33 |
-
|
|
|
|
| 34 |
|
| 35 |
# =========================
|
| 36 |
# 1. LOAD TOKENIZER
|
|
|
|
| 30 |
# SMALL LLM (POST-EDITOR)
|
| 31 |
# ============================================================
|
| 32 |
|
| 33 |
+
|
| 34 |
+
LLM = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
|
| 35 |
|
| 36 |
# =========================
|
| 37 |
# 1. LOAD TOKENIZER
|