AlbertoFH98 commited on
Commit
627a100
1 Parent(s): f22abe1

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -1
utils.py CHANGED
@@ -115,7 +115,7 @@ PREGUNTA:""", cleaned_prompt, re.DOTALL)
115
  return text, new_cleaned_prompt
116
 
117
  # -- Get GPT response
118
- def get_gpt_response(query):
119
  template = """Eres un asistente. Su misión es proporcionar respuestas precisas a preguntas relacionadas con la transcripción de una entrevista de YouTube.
120
  No saludes en tu respuesta. No repita la pregunta en su respuesta. Sea conciso y omita las exenciones de responsabilidad o los mensajes predeterminados.
121
  Solo responda la pregunta, no agregue texto adicional. No des tu opinión personal ni tu conclusión personal. No haga conjeturas ni suposiciones.
@@ -125,6 +125,7 @@ def get_gpt_response(query):
125
  Respuesta:"""
126
 
127
  rag_prompt_custom = PromptTemplate.from_template(template)
 
128
  docs = loader.load()
129
 
130
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
 
115
  return text, new_cleaned_prompt
116
 
117
  # -- Get GPT response
118
+ def get_gpt_response(transcription_path, query):
119
  template = """Eres un asistente. Su misión es proporcionar respuestas precisas a preguntas relacionadas con la transcripción de una entrevista de YouTube.
120
  No saludes en tu respuesta. No repita la pregunta en su respuesta. Sea conciso y omita las exenciones de responsabilidad o los mensajes predeterminados.
121
  Solo responda la pregunta, no agregue texto adicional. No des tu opinión personal ni tu conclusión personal. No haga conjeturas ni suposiciones.
 
125
  Respuesta:"""
126
 
127
  rag_prompt_custom = PromptTemplate.from_template(template)
128
+ loader = TextLoader(transcription_path)
129
  docs = loader.load()
130
 
131
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)