Update app.py

app.py

@@ -1,6 +1,10 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
+import spaces
+
+
+
 
 # 1. تحميل النموذج
 model_id = "yasserrmd/kallamni-1.2b-v1"
@@ -27,6 +31,7 @@ few_shot = [
 ]
 
 # 3. دالة التوليد
+@spaces.GPU
 def chat_fn(message, history):
     try:
         # Gradio ChatInterface history = list of dicts [{"role":..., "content":...}]