import streamlit as st from transformers import pipeline from ModelDriver import RobertaSentinelOpenGPTInference, RobertaSentinelCSAbstractInference # Add a title st.title('GPT Detection Demo') # Add 4 options for 4 models option = st.sidebar.selectbox( 'Which Model do you want to use?', ('RobertaSentinelOpenGPT', 'RobertaSentinelCSAbstract'), ) text = st.text_area('Enter text here', '') if st.button('Generate'): if option == 'RobertaSentinelOpenGPT': result = RobertaSentinelOpenGPTInference(text) elif option == 'RobertaSentinelCSAbstract': result = RobertaSentinelCSAbstractInference(text) st.write(result)