Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
+import spaces
 
 model_id = "aeonium/Aeonium-v1.1-Chat-4B"
 
@@ -12,6 +13,7 @@ model = AutoModelForCausalLM.from_pretrained(
     torch_dtype=torch.bfloat16,
 )
 
+@spaces.GPU
 def predict(message, history, max_tokens, temperature, top_p):
     # Формирование чата из истории и нового сообщения
     chat = [{"role": "user" if i % 2 == 0 else "assistant", "content": m} 
@@ -45,12 +47,6 @@ iface = gr.ChatInterface(
     title="Чат с Aeonium v1.1",
     description="Это чат-интерфейс для модели Aeonium v1.1 Chat 4B. Задавайте вопросы и получайте ответы!",
     theme="soft",
-    examples=[
-        "Привет! Как дела?",
-        "Расскажи мне о квантовой физике",
-        "Какие интересные книги ты можешь порекомендовать?",
-    ],
-    cache_examples=False,
     retry_btn="Повторить",
     undo_btn="Отменить последнее",
     clear_btn="Очистить",