Update app.py

app.py

@@ -3,7 +3,7 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import random
 
 # Загрузка модели и токенизатора
-model_name = "ai-forever/rugpt3small_based_on_gpt2"
+model_name = "ai-forever/rugpt3large_based_on_gpt2"
 tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 model = GPT2LMHeadModel.from_pretrained(model_name)
 
@@ -16,9 +16,11 @@ def generate(prompt, _=None):
     random_phrase = random.choice(random_phrases)
     user_input = f"[USER]: {prompt}\n[BOT]: {random_phrase},"
     inputs = tokenizer.encode(user_input, return_tensors="pt")
-    outputs = model.generate(inputs, max_length=50, num_return_sequences=1, pad_token_id=tokenizer.eos_token_id)
+    outputs = model.generate(inputs, max_length=25, num_return_sequences=1, pad_token_id=tokenizer.eos_token_id)
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
     edit_response = response.split(f"[BOT]: ")[-1].strip()
+    if "[" in edit_response:
+      edit_response = edit_response.split("[")[0]
     bot_message = f"[BOT!]: {edit_response}"
     return bot_message