dtrejopizzo commited on
Commit
826d387
1 Parent(s): 6b7eaf7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -27,7 +27,9 @@ def construct_index(directory_path):
27
 
28
  def chatbot(input_text):
29
  # load the index from disk
30
- index = GPTVectorStoreIndex.load_from_disk('index.json')
 
 
31
  # define custom Prompt
32
  TEMPLATE_STR = (
33
  "Quiero que actues como un asistente personal de un cliente del Banco Galicia. Tu nombre es Gala. Me brindas informaci贸n sobre mi resument de tarjeta de credito VISA. Si la respuesta no esta en el documento, respondeme de forma creativa que no lo sabes, pero que podes ayudarme con otra pregunta. Nunca te enojes y no contestes preguntas politicas o religiosas. \n"
@@ -36,7 +38,7 @@ def chatbot(input_text):
36
  )
37
  QA_TEMPLATE = PromptTemplate(TEMPLATE_STR)
38
  # query the index and get the response
39
- response = query_engine.query(QA_TEMPLATE)
40
  return response.response
41
 
42
  iface = gr.Interface(fn=chatbot,
 
27
 
28
  def chatbot(input_text):
29
  # load the index from disk
30
+ storage_context = StorageContext.from_defaults(persist_dir="index.json")
31
+ # load index
32
+ index = load_index_from_storage(storage_context)
33
  # define custom Prompt
34
  TEMPLATE_STR = (
35
  "Quiero que actues como un asistente personal de un cliente del Banco Galicia. Tu nombre es Gala. Me brindas informaci贸n sobre mi resument de tarjeta de credito VISA. Si la respuesta no esta en el documento, respondeme de forma creativa que no lo sabes, pero que podes ayudarme con otra pregunta. Nunca te enojes y no contestes preguntas politicas o religiosas. \n"
 
38
  )
39
  QA_TEMPLATE = PromptTemplate(TEMPLATE_STR)
40
  # query the index and get the response
41
+ response = index.query(QA_TEMPLATE)
42
  return response.response
43
 
44
  iface = gr.Interface(fn=chatbot,