Update LLMPipeline.py

LLMPipeline.py

@@ -78,9 +78,9 @@ def generate_image_prompt_viejo(summary):
     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=400, do_sample=True, temperature=0.7)[0]['generated_text']
     return result.strip()
-from transformers import T5Tokenizer, T5ForConditionalGeneration
-tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-small")
-model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-small")
+# from transformers import T5Tokenizer, T5ForConditionalGeneration
+tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-small")
+model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-small")
 
 def generate_image_prompt(summary):
     input_text = f"Create a prompt for image generation. Should be a meme. Short, visual and funny. Related to: {summary}"