dtrejopizzo commited on
Commit
3ef9183
1 Parent(s): 2649f54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  from llama_index import Prompt, SimpleDirectoryReader, LLMPredictor, PromptHelper, StorageContext, ServiceContext, GPTVectorStoreIndex, load_index_from_storage
2
  from langchain.chat_models import ChatOpenAI
3
- from langchain.prompts import PromptTemplate
4
  import gradio as gr
5
  import sys
6
  import os
@@ -36,7 +36,10 @@ def chatbot(input_text):
36
  #QA_TEMPLATE = PromptTemplate(TEMPLATE_STR)
37
  # query the index and get the response
38
  # Instantiation using initializer
39
- prompt = PromptTemplate(input_variables=input_text, template="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. Dado esto, por favor responde a la pregunta: {input_text}")
 
 
 
40
  response = query_engine.query(prompt)
41
  return response.response
42
 
 
1
  from llama_index import Prompt, SimpleDirectoryReader, LLMPredictor, PromptHelper, StorageContext, ServiceContext, GPTVectorStoreIndex, load_index_from_storage
2
  from langchain.chat_models import ChatOpenAI
3
+ from langchain.prompts import StringPromptTemplate
4
  import gradio as gr
5
  import sys
6
  import os
 
36
  #QA_TEMPLATE = PromptTemplate(TEMPLATE_STR)
37
  # query the index and get the response
38
  # Instantiation using initializer
39
+ prompt = """\
40
+ 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. Dado esto, por favor responde a la pregunta: {input_text}")
41
+ """
42
+ #prompt = PromptTemplate(input_variables=input_text, template="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. Dado esto, por favor responde a la pregunta: {input_text}")
43
  response = query_engine.query(prompt)
44
  return response.response
45