Update LLMPipeline.py

LLMPipeline.py

@@ -61,21 +61,21 @@ text_generator = pipeline("text-generation", model=model, tokenizer=tokenizer, t
 # Funciones
 # ----------------------------
 
-def summarize_headlines(headlines):
-    """
-    rewrites the headings in a humoristic way
-    """
-    prompt = "Rewrite: " + "\n".join(headlines) + "Humoristic version. Short. Just one phrase"
-    result = text_generator(prompt, max_new_tokens=50, do_sample=True, temperature=0.7, return_full_text=False)
-    # print('resultadoooo: ', result)
-    result= result[0]['generated_text']
-    return result.strip()
+# def summarize_headlines(headlines):
+#     """
+#     rewrites the headings in a humoristic way
+#     """
+#     prompt = "Rewrite: " + "\n".join(headlines) + "Humoristic version. Short. Just one phrase"
+#     result = text_generator(prompt, max_new_tokens=50, do_sample=True, temperature=0.7, return_full_text=False)
+#     # print('resultadoooo: ', result)
+#     result= result[0]['generated_text']
+#     return result.strip()
 
 def generate_image_prompt(summary):
     """
     Genera un prompt corto y visual para un generador de imágenes tipo meme.
     """
-    prompt = f"Crea un prompt para imagen tipo meme sobre este resumen de noticias:\n{summary}\nEl prompt debe ser corto, visual y gracioso. Devuelve solo el prompt."
+    prompt = f"Crea un prompt para imagen tipo meme sobre este titular de noticias:\n{summary}\nEl prompt debe ser corto, visual y gracioso. Devuelve solo el prompt."
     result = text_generator(prompt, max_new_tokens=50, do_sample=True, temperature=0.7, return_full_text=False)[0]['generated_text']
     return result.strip()