Update app.py

app.py

@@ -89,12 +89,17 @@ from langchain_core.runnables import RunnableLambda
 #prompt_template = ChatPromptTemplate.from_template(prompt)
 llm = HuggingFaceHub(
     repo_id="HuggingFaceH4/zephyr-7b-beta",
-    model_kwargs={"temperature": 0.1, "max_length": 10, "return_full_text" : False}
+    model_kwargs={
+        "temperature": 0.1,
+        "max_length": 10,
+        "return_full_text": False,
+        "stop_sequences": ["Explanation:", "Original question:", "\n"]
+    }
 )
 
 def ra(user_question):
     #prompt = f"You know things about League of Legends. Please correct the following question for grammar and clarity.Do not give explaination:\n{user_question}\nCorrected question:"
-    prompt = f"Correct the following question for grammar and clarity. Output only the corrected question. Do not explain.\n\n{user_question}\n\nCorrected question:"
+    prompt = f"Rewrite the following question with correct grammar. ONLY return the corrected question. Do NOT include any explanation.\n\n{user_question}"
 
     # Pass the prompt to the LLM and get the response
     rephrased_query = llm(prompt)  # Replace `llm` with the appropriate LLM function or API call