sgt444pepper commited on
Commit
3a30f27
·
verified ·
1 Parent(s): 4e8afdd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -5
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()