Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import torch
 import os
+from spaces import GPU
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 # Configuración del modelo
@@ -11,7 +12,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id).to(device)
 
 
-# Función de respuesta del chatbot
+@GPU
 def respond(message, history, system_message, max_tokens, temperature, top_p):
     prompt = system_message + "\n"
     for user, assistant in history:
@@ -35,6 +36,7 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
 
 
 
+
 # Descripción del proyecto
 descripcion = """
 
@@ -125,3 +127,4 @@ demo.launch(
 
 
 
+