Update llm.py

llm.py

@@ -7,7 +7,8 @@ from langchain_community.llms.huggingface_pipeline import HuggingFacePipeline
 # Initialize HF pipeline for text generation
 text_generator = pipeline(
     "text-generation",  # Task type
-    model="google/gemma-3n-e4b-it",
+    model="google/gemma-3n-E2B-it-litert-preview"
+    # model="google/gemma-3n-e4b-it",
     # model="Qwen/Qwen3-Embedding-0.6B",
     # device="cuda" if torch.cuda.is_available() else "cpu",
     device= "cpu",
@@ -20,16 +21,9 @@ model = HuggingFacePipeline(pipeline=text_generator)
 
 def generate_sentences(topic, n=1):
     prompt = ChatPromptTemplate.from_template(
-        "### Instruction\n"
-        "Create exactly {n} English sentences about {topic} with these rules:\n"
-        "- Maximum 10 words per sentence\n"
-        "- Use simple vocabulary\n"
-        "- Family-friendly content\n\n"
-        "### Output Format\n"
-        "Return ONLY the sentences, one per line with:\n"
-        "- No bullet points\n"
-        "- No numbering\n"
-        "- No extra text or explanations"
+        "You are a helpful assistant. Generate exactly {n} simple sentences about the topic: {topic}. "
+        "Each sentence must be in English and appropriate for all audiences. "
+        "Return each sentence on a new line without any numbering or bullets"
     )
     chain = prompt | model | StrOutputParser()
     response = chain.invoke({"topic": topic, "n": n})