mhsvieira commited on
Commit
3e66511
1 Parent(s): 2b3e58c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -102,6 +102,7 @@ def main():
102
  st.warning(st.session_state['msg'])
103
  msg = 'Prosseguir' if portuguese else 'Proceed'
104
  if st.button(msg):
 
105
  text = extract(query, search_model=search_model, extracted_documents=st.session_state['documents'])
106
  summary = summarize(text, summ_model, tokenizer)
107
 
 
102
  st.warning(st.session_state['msg'])
103
  msg = 'Prosseguir' if portuguese else 'Proceed'
104
  if st.button(msg):
105
+ query = translate(query_pt, 'pt', 'en') if portuguese else query
106
  text = extract(query, search_model=search_model, extracted_documents=st.session_state['documents'])
107
  summary = summarize(text, summ_model, tokenizer)
108