Kukedlc commited on
Commit
2cbff7e
·
verified ·
1 Parent(s): 886c73a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -15,6 +15,7 @@ ruta_modelo = "01-ai/Yi-9B-Chat"
15
  tokenizador = AutoTokenizer.from_pretrained(ruta_modelo)
16
  modelo = AutoModelForCausalLM.from_pretrained(ruta_modelo, device_map="auto").eval()
17
 
 
18
  def generar_respuesta(historial, usuario_input, max_longitud):
19
  mensajes = [
20
  {"role": "system", "content": "Eres un asistente útil y amigable. Proporciona respuestas claras y concisas."}
 
15
  tokenizador = AutoTokenizer.from_pretrained(ruta_modelo)
16
  modelo = AutoModelForCausalLM.from_pretrained(ruta_modelo, device_map="auto").eval()
17
 
18
+ @spaces.GPU(duration=130)
19
  def generar_respuesta(historial, usuario_input, max_longitud):
20
  mensajes = [
21
  {"role": "system", "content": "Eres un asistente útil y amigable. Proporciona respuestas claras y concisas."}