BroBro87 commited on
Commit
fb333f9
1 Parent(s): fb05b67

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -40,6 +40,7 @@ def initialize_vector_store_index(data_path, service_context):
40
  return index
41
 
42
  # Configure and initialize components
 
43
  llm = configure_llama_model()
44
  embed_model = configure_embeddings()
45
  service_context = configure_service_context(llm, embed_model)
@@ -50,7 +51,11 @@ query_engine = index.as_query_engine()
50
  def get_response(text, username):
51
  # For simplicity, we are only using the 'text' argument
52
  response = str(query_engine.query(text))
53
- return response
 
 
 
 
54
 
55
 
56
- gr.ChatInterface(get_response).launch(debug=True)
 
40
  return index
41
 
42
  # Configure and initialize components
43
+ dit = {}
44
  llm = configure_llama_model()
45
  embed_model = configure_embeddings()
46
  service_context = configure_service_context(llm, embed_model)
 
51
  def get_response(text, username):
52
  # For simplicity, we are only using the 'text' argument
53
  response = str(query_engine.query(text))
54
+ if text not in dit.keys():
55
+ dit[text]=response
56
+ return response
57
+ else:
58
+ return dit[text]
59
 
60
 
61
+ gr.ChatInterface(get_response).launch(debug=True,share=True)