SimrusDenuvo commited on
Commit
eaa75b3
·
verified ·
1 Parent(s): 2281253

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
5
  import torch
6
 
7
  # Используем модель Flan-T5 или Sberbank AI
8
- model_name = "cointegrated/rut5-base" # Можно заменить на sberbank-ai/rugpt3small или другую
9
 
10
  tokenizer = AutoTokenizer.from_pretrained(model_name)
11
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
5
  import torch
6
 
7
  # Используем модель Flan-T5 или Sberbank AI
8
+ model_name = "cointegrated/rugpt2-large" # Можно заменить на sberbank-ai/rugpt3small или другую
9
 
10
  tokenizer = AutoTokenizer.from_pretrained(model_name)
11
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)