Update app.py

app.py

@@ -76,8 +76,13 @@ qa_chat_prompt = ChatPromptTemplate.from_messages(
 )
 
 llm_model = "bigscience/bloom-560m"
+from transformers import AutoTokenizer
+tokenizer = AutoTokenizer.from_pretrained(llm_model)
+
 print("check2")
-pipe = pipeline(model = llm_model, tokenizer = retriever, task="text-generation",chat = qa_chat_prompt)
+pipe = pipeline(model = llm_model, tokenizer = tokenizer, task="text-generation",chat = qa_chat_prompt)
+#"question-answering", "conversational"
+
 print("check3")
 
 #chain = qa_chat_prompt | pipe