Update app.py

app.py

@@ -2,7 +2,10 @@ import requests
 from bs4 import BeautifulSoup
 import gradio as gr
 from langchain.chains.question_answering import load_qa_chain
-from langchain.llms import HuggingFaceHub
+# from langchain.llms import HuggingFaceHub
+from transformers import pipeline
+from langchain.llms import HuggingFacePipeline
+from langchain.chains.question_answering import load_qa_chain
 from transformers import pipeline
 from sentence_transformers import SentenceTransformer
 from sklearn.metrics.pairwise import cosine_similarity
@@ -29,7 +32,14 @@ translate_ru_to_en = pipeline("translation", model="Helsinki-NLP/opus-mt-ru-en")
 
 
 # Russian Language Model for QA
-llm = HuggingFaceHub(repo_id="DeepPavlov/rubert-base-cased", model_kwargs={"temperature": 0})
+#llm = HuggingFaceHub(repo_id="DeepPavlov/rubert-base-cased", model_kwargs={"temperature": 0})
+# Create a QA pipeline using the DeepPavlov model directly
+qa_pipeline = pipeline(
+    "question-answering",
+    model="DeepPavlov/rubert-base-cased",
+    tokenizer="DeepPavlov/rubert-base-cased"
+)
+llm = HuggingFacePipeline(pipeline=qa_pipeline)
 qa_chain = load_qa_chain(llm, chain_type="stuff")
 
 # Embedding Model