Spaces:
Sleeping
Sleeping
Upload 2 files
Browse files
app.py
CHANGED
@@ -65,7 +65,10 @@ def generar_respuesta(pregunta):
|
|
65 |
).to("cpu")
|
66 |
|
67 |
with torch.no_grad():
|
68 |
-
outputs = model.generate(
|
|
|
|
|
|
|
69 |
|
70 |
return tokenizer.decode(outputs[0], skip_special_tokens=True)
|
71 |
|
|
|
65 |
).to("cpu")
|
66 |
|
67 |
with torch.no_grad():
|
68 |
+
outputs = model.generate(
|
69 |
+
**inputs,
|
70 |
+
max_new_tokens=100 # Generar hasta 100 tokens adicionales
|
71 |
+
)
|
72 |
|
73 |
return tokenizer.decode(outputs[0], skip_special_tokens=True)
|
74 |
|