Update app.py

app.py

@@ -8,7 +8,7 @@ import time
 from transformers import AutoTokenizer, AutoModelForQuestionAnswering
 import torch
 
-retriever = load_the_embedding_retrieve(is_ready=True, k=3)
+retriever = load_the_embedding_retrieve(is_ready=False, k=3)
 bm25_retriever = load_the_bm25_retrieve(k=3)
 
 ensemble_retriever = EnsembleRetriever(
@@ -20,13 +20,13 @@ model = AutoModelForQuestionAnswering.from_pretrained("ShynBui/vie_qa", token=os
 
 headers = {
     "Accept": "application/json",
-    "Authorization": "Bearer "+ os.environ.get("HF_TOKEN"),
+    "Authorization": "Bearer " + os.environ.get("HF_TOKEN"),
     "Content-Type": "application/json"
 }
 
 
 def query(payload):
-    response = requests.post(API_URL, headers=headers, json=payload)
+    response = requests.post(os.environ.get("API_URL"), headers=headers, json=payload)
     return response.json()
 
 
@@ -62,6 +62,7 @@ def greet(quote):
         answers.append(output['answer'])
     return answers
 
+
 def greet2(quote):
     answers = []
     docs = ensemble_retriever.get_relevant_documents(quote)
@@ -86,7 +87,7 @@ def greet2(quote):
 
 
 if __name__ == "__main__":
-
+    quote = "Địa chỉ nhà trường?"
 
     iface = gr.Interface(fn=greet2, inputs="text", outputs="text")
     iface.launch()