Update app.py

app.py

@@ -92,8 +92,7 @@ llm = HuggingFaceHub(
     model_kwargs={
         "temperature": 0.1,
         "max_length": 10,
-        "return_full_text": False,
-        "stop_sequences": ["Explanation:", "Original question:", "\n"]
+        "return_full_text": False
     }
 )
 
@@ -102,7 +101,7 @@ def ra(user_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
+    rephrased_query = llm.invoke(prompt)  # Replace `llm` with the appropriate LLM function or API call
     new_query = rephrased_query.strip()
     return {'context' : retriever(new_query), 'question': new_query}
 # chain = RunnablePassthrough() | RunnableLambda(ra) | prompt_template | client.chat_completion() | StrOutputParser() for notebook