Update app.py

app.py

@@ -32,7 +32,8 @@ def initialize_chatbot(vector_db):
         repo_id="mistralai/Mistral-7B-Instruct-v0.2",
         huggingfacehub_api_token=os.environ.get("HUGGINGFACE_API_TOKEN"),
         temperature=0.5,
-        max_new_tokens=256
+        max_new_tokens=256,
+        task="text-generation"  # Explicitly specify the task type
     )
     qa_chain = ConversationalRetrievalChain.from_llm(
         llm=llm,