import streamlit as st
from transformers import pipeline

# Title for the app
st.title('Fill in the Blank Bot')

# Load the fill-mask pipeline
fill_mask = pipeline('fill-mask')

# Text area for the user input
user_input = st.text_area("Enter a sentence with a <mask> in place of the missing word")

# Button to trigger the fill-mask function
if st.button('Fill the Blank'):
    # Predict the missing words
    results = fill_mask(user_input)

    # Display the top 5 predictions
    st.write("Top predictions:")
    for result in results:
        st.write(f"Prediction: {result['token_str']}, Confidence: {round(result['score'], 4)}")