Update LLMPipeline.py

LLMPipeline.py

@@ -78,9 +78,10 @@ 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
+
 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}"
     input_ids = tokenizer(input_text, return_tensors="pt").input_ids