app.py

"""
# My first app
Here's our first attempt at using data to create a table:
"""
import streamlit as st
import pandas as pd
from load_css import local_css
from retriever import do_search

local_css('style.css')

st.header('Semantic search demo')
search = st.text_input('')

if search:
    result = do_search(search)
    col1, col2, col3 = st.columns(3)

    with col1:
        st.write('TF-IDF')
        st.write(result[0])

    with col2:
        st.write('Base dense retriever')
        st.write(result[1])

    with col3:
        st.write('Adapted dense retriever')
        st.write(result[2])