Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 import os
-
+import spaces
 # === CONFIGURATION ===
 MODEL_NAME = "facebook/MobileLLM-Pro"   # ✅ Plus léger et rapide sur CPU
 INFO_FILE = "infos_medicaux.txt"
@@ -27,7 +27,7 @@ else:
     medical_context = ""
     print("⚠️ Aucun fichier infos_medicaux.txt trouvé.")
 
-# === CHATBOT ===
+@spaces.GPU()
 def chat_with_finanfa(message, history=None):
     if history is None:
         history = []