QAway-to
commited on
Commit
·
62409b9
1
Parent(s):
7fa6779
New model Google-gemma-1.1-2b-it. v1.1
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification, Auto
|
|
| 7 |
# 1. Настройка моделей
|
| 8 |
# =========================================================
|
| 9 |
MBTI_MODEL_ID = "f3nsmart/MBTIclassifier"
|
| 10 |
-
LLM_MODEL_ID = "google/gemma-2b-it" # Быстрая, контекстная и лаконичная
|
| 11 |
|
| 12 |
# Классификатор (твой fine-tuned)
|
| 13 |
mbti_tokenizer = AutoTokenizer.from_pretrained(MBTI_MODEL_ID)
|
|
|
|
| 7 |
# 1. Настройка моделей
|
| 8 |
# =========================================================
|
| 9 |
MBTI_MODEL_ID = "f3nsmart/MBTIclassifier"
|
| 10 |
+
LLM_MODEL_ID = "google/gemma-1.1-2b-it" # Быстрая, контекстная и лаконичная
|
| 11 |
|
| 12 |
# Классификатор (твой fine-tuned)
|
| 13 |
mbti_tokenizer = AutoTokenizer.from_pretrained(MBTI_MODEL_ID)
|