Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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."}
|