Update app.py

app.py

@@ -1,18 +1,14 @@
 from haystack.document_stores.memory import InMemoryDocumentStore
 from haystack.nodes import TfidfRetriever, FARMReader
-import google.colab
-google.colab.drive.mount('/content/drive')
 
 import pickle
 
-pickle_file = '/content/drive/MyDrive/Group13_NLP_Project/knowledge_graph.pickle'
+pickle_file = 'knowledge_graph.pickle'
 
 # Load the knowledge graph from the pickle file
 with open(pickle_file, 'rb') as f:
   knowledge_graph = pickle.load(f)
 
-print("Knowledge graph loaded from ", pickle_file)
-
 document_store = InMemoryDocumentStore()
 node_sentences = {}
 documents = []
@@ -47,9 +43,11 @@ document_store.write_documents(documents)
 
 #Initialize the retriever
 retriever = TfidfRetriever(document_store=document_store)
+
 #Initialize the reader
 model_name = "primasr/multilingualbert-for-eqa-finetuned"
 reader = FARMReader(model_name_or_path=model_name, use_gpu=False)
+
 #Create pipeline with the component of retriever and reader
 from haystack.pipelines import Pipeline
 pipeline = Pipeline()
@@ -96,11 +94,8 @@ def checkReiterateQuery(query,lang):
 
 import gradio as gr
 from langdetect import detect
-import warnings
-warnings.filterwarnings('ignore')
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-
 chat_history = []
 answer_counter = 0
 def chatbot_interface(message):