Update app.py

app.py

@@ -9,12 +9,12 @@ def test_gpt_simple():
         print("Initializing text generation pipeline...")
         generator = pipeline(
             'text-generation',
-            model='ai-forever/rugpt3medium_based_on_gpt2',
+            model='ai-forever/rugpt3small_based_on_gpt2',
             device=0 if torch.cuda.is_available() else -1
         )
         
         # Загружаем токенизатор
-        tokenizer = AutoTokenizer.from_pretrained('ai-forever/rugpt3medium_based_on_gpt2')
+        tokenizer = AutoTokenizer.from_pretrained('ai-forever/rugpt3small_based_on_gpt2')
         
         test_prompt = """Текст: 
 [Текст песни "Имя моей тоски"]