Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -59,21 +59,25 @@ def summarize(input_text, file, summary_type, method, num_sentences, model_name,
|
|
| 59 |
elif summary_type == "Combinado":
|
| 60 |
if model_name not in summarizers:
|
| 61 |
return "Modelo no disponible para resumen abstractivo."
|
| 62 |
-
|
| 63 |
-
# Generar
|
| 64 |
-
|
| 65 |
preprocessor.split_into_sentences(cleaned_text),
|
| 66 |
preprocessor.clean_sentences(preprocessor.split_into_sentences(cleaned_text)),
|
| 67 |
num_sentences,
|
| 68 |
)
|
|
|
|
|
|
|
|
|
|
| 69 |
return handle_long_text(
|
| 70 |
-
|
| 71 |
-
summarizers[model_name][0],
|
| 72 |
-
summarizers[model_name][1],
|
| 73 |
max_length=max_length,
|
| 74 |
stride=128,
|
| 75 |
)
|
| 76 |
|
|
|
|
| 77 |
return "Seleccione un tipo de resumen válido."
|
| 78 |
|
| 79 |
# Interfaz dinámica
|
|
|
|
| 59 |
elif summary_type == "Combinado":
|
| 60 |
if model_name not in summarizers:
|
| 61 |
return "Modelo no disponible para resumen abstractivo."
|
| 62 |
+
|
| 63 |
+
# Paso 1: Generar puntos clave con el método extractivo
|
| 64 |
+
extractive_keypoints = TFIDFSummarizer().summarize(
|
| 65 |
preprocessor.split_into_sentences(cleaned_text),
|
| 66 |
preprocessor.clean_sentences(preprocessor.split_into_sentences(cleaned_text)),
|
| 67 |
num_sentences,
|
| 68 |
)
|
| 69 |
+
|
| 70 |
+
# Paso 2: Generar el resumen abstractivo guiado por los puntos clave
|
| 71 |
+
combined_input = f"{cleaned_text}\n\nPuntos clave: {extractive_keypoints}"
|
| 72 |
return handle_long_text(
|
| 73 |
+
combined_input, # Texto completo + puntos clave
|
| 74 |
+
summarizers[model_name][0], # Modelo abstractivo seleccionado
|
| 75 |
+
summarizers[model_name][1], # Tokenizador del modelo abstractivo
|
| 76 |
max_length=max_length,
|
| 77 |
stride=128,
|
| 78 |
)
|
| 79 |
|
| 80 |
+
|
| 81 |
return "Seleccione un tipo de resumen válido."
|
| 82 |
|
| 83 |
# Interfaz dinámica
|