Addind Chat functions to QA retrival chain

app.py

@@ -61,10 +61,18 @@ def youtube_chat(youtube_link,API_key,llm='HuggingFace',temperature=0.1,max_toke
     global vector_db
     vector_db = FAISS.from_documents(documents=texts, embedding= embedding_model)
     global qa
-    qa = RetrievalQA.from_chain_type(llm=chatops.chat_application(llm_service=llm,key=API_key,
-                                                            temperature=temperature,
-                                                            max_tokens=max_tokens
-                                                        ),
+
+    if llm == 'HuggingFace':
+        chat = chatops.get_hugging_face_model(
+                            model_id="tiiuae/falcon-7b-instruct",
+                            key=API_key,
+                            temperature=temperature,
+                            max_tokens=max_tokens
+                            )
+    else:
+        chat = chatops.get_openai_chat_model(API_key=API_key)
+
+    qa = RetrievalQA.from_chain_type(llm=chat,
                                 chain_type='stuff',
                                 retriever=vector_db.as_retriever(),
                                 #  chain_type_kwargs=chain_type_kwargs,