lfoppiano commited on
Commit
8913fe2
·
1 Parent(s): 07c3089

fix annotation being null

Browse files
Files changed (1) hide show
  1. streamlit_app.py +1 -2
streamlit_app.py CHANGED
@@ -6,7 +6,6 @@ from tempfile import NamedTemporaryFile
6
  import dotenv
7
  from grobid_quantities.quantities import QuantitiesAPI
8
  from langchain.memory import ConversationBufferWindowMemory
9
- # from langchain_community.callbacks import PromptLayerCallbackHandler
10
  from langchain_community.chat_models import ChatOpenAI
11
  from langchain_community.llms.huggingface_endpoint import HuggingFaceEndpoint
12
  from langchain_huggingface import HuggingFaceEmbeddings
@@ -499,7 +498,7 @@ with left_column:
499
  pdf_viewer(
500
  input=st.session_state['binary'],
501
  annotation_outline_size=2,
502
- annotations=st.session_state['annotations'],
503
  render_text=True,
504
  scroll_to_annotation=1 if (st.session_state['annotations'] and st.session_state['scroll_to_first_annotation']) else None
505
  )
 
6
  import dotenv
7
  from grobid_quantities.quantities import QuantitiesAPI
8
  from langchain.memory import ConversationBufferWindowMemory
 
9
  from langchain_community.chat_models import ChatOpenAI
10
  from langchain_community.llms.huggingface_endpoint import HuggingFaceEndpoint
11
  from langchain_huggingface import HuggingFaceEmbeddings
 
498
  pdf_viewer(
499
  input=st.session_state['binary'],
500
  annotation_outline_size=2,
501
+ annotations=st.session_state['annotations'] if st.session_state['annotations'] else [],
502
  render_text=True,
503
  scroll_to_annotation=1 if (st.session_state['annotations'] and st.session_state['scroll_to_first_annotation']) else None
504
  )