Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -113,18 +113,26 @@ class QAChatbot:
|
|
113 |
)
|
114 |
return response
|
115 |
|
|
|
116 |
def chatbot_interface(query, history):
|
117 |
-
file_paths = ["Company_eng.txt", "base_eng.txt"]
|
118 |
-
documents = load_documents(file_paths)
|
119 |
|
120 |
-
indexer = Retriver(documents, tokenizer, model)
|
121 |
-
reranker = Reranker(reranker_model)
|
122 |
|
123 |
-
chatbot = QAChatbot(indexer, reranker)
|
124 |
answer = chatbot.generate_answer(query)
|
125 |
return answer["choices"][0]["message"]["content"]
|
126 |
|
127 |
iface = gr.ChatInterface(fn=chatbot_interface, type="messages")
|
128 |
|
129 |
if __name__ == "__main__":
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
130 |
iface.launch()
|
|
|
113 |
)
|
114 |
return response
|
115 |
|
116 |
+
|
117 |
def chatbot_interface(query, history):
|
118 |
+
# file_paths = ["Company_eng.txt", "base_eng.txt"]
|
119 |
+
# documents = load_documents(file_paths)
|
120 |
|
121 |
+
# indexer = Retriver(documents, tokenizer, model)
|
122 |
+
# reranker = Reranker(reranker_model)
|
123 |
|
124 |
+
#chatbot = QAChatbot(indexer, reranker)
|
125 |
answer = chatbot.generate_answer(query)
|
126 |
return answer["choices"][0]["message"]["content"]
|
127 |
|
128 |
iface = gr.ChatInterface(fn=chatbot_interface, type="messages")
|
129 |
|
130 |
if __name__ == "__main__":
|
131 |
+
file_paths = ["Company_eng.txt", "base_eng.txt"]
|
132 |
+
documents = load_documents(file_paths)
|
133 |
+
|
134 |
+
indexer = Retriver(documents, tokenizer, model)
|
135 |
+
reranker = Reranker(reranker_model)
|
136 |
+
|
137 |
+
chatbot = QAChatbot(indexer, reranker)
|
138 |
iface.launch()
|