SimrusDenuvo commited on
Commit
d5e56ca
·
verified ·
1 Parent(s): 1382bb2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -7,15 +7,15 @@ from datasets import load_dataset
7
  dataset = load_dataset("ZhenDOS/alpha_bank_data")
8
 
9
  # Инициализация разных моделей
10
- MODELS = {
11
- "GigaChat-like": "ai-forever/rugpt3large_based_on_gpt2", # Русская модель большого размера
12
  "ChatGPT-like": "tinkoff-ai/ruDialoGPT-medium", # Диалоговая модель для русского языка
13
  "DeepSeek-like": "ai-forever/sbert_large_nlu_ru" # Русская модель для понимания текста
14
- }
15
 
16
  # Инициализация моделей и токенизаторов
17
- models = {}
18
- tokenizers = {}
19
 
20
  for model_name, model_path in MODELS.items():
21
  try:
 
7
  dataset = load_dataset("ZhenDOS/alpha_bank_data")
8
 
9
  # Инициализация разных моделей
10
+ model_name = "ai-forever/rugpt3large_based_on_gpt2"
11
+ "GigaChat-like:" "ai-forever/rugpt3large_based_on_gpt2", # Русская модель большого размера
12
  "ChatGPT-like": "tinkoff-ai/ruDialoGPT-medium", # Диалоговая модель для русского языка
13
  "DeepSeek-like": "ai-forever/sbert_large_nlu_ru" # Русская модель для понимания текста
14
+
15
 
16
  # Инициализация моделей и токенизаторов
17
+ models = model_name
18
+ tokenizers = AutoTokenizer.from_pretrained(model_name)
19
 
20
  for model_name, model_path in MODELS.items():
21
  try: