model changed

app.py

@@ -3,7 +3,7 @@ from textPreprocessing import text2prompt
 from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
 import torch
 
-model_id = 'openlm-research/open_llama_3b_v2'
+model_id = 'ai-forever/ruT5-large'
 
 """bnb_config = BitsAndBytesConfig(
     load_in_4bit=True,

textPreprocessing.py

@@ -5,7 +5,7 @@ def text2prompt(text: str):
   """
 
   instruction = 'Ниже тебе дан текст, содержащий новость на тему экономики. ' \
-     'На основе текста ты должен дать наиболее подходящую финансовую рекомендацию из списка ["buy", "sell", "long", "short"]. ' \
+     'На основе текста ты должен дать наиболее подходящую финансовую рекомендацию из списка ["купить", "продать", "долгая позиция", "короткия позиция"]. ' \
      'Твой ответ должен содержать только одно слово, которое будет подходящей рекомендацией из списка.'
 
   bos_token = "<s>"