Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 from setfit import SetFitModel
 
 # Load the model
-model = SetFitModel.from_pretrained("peter2000/vulnerable-groups-setfit")
+model = SetFitModel.from_pretrained("leavoigt/vulnerable-groups")
 
 # Define the classes
 group_dict = {
@@ -49,7 +49,10 @@ st.write("This app allows you to identify whether a text contains any references
 
 col1, col2 = st.columns(2)
 
-col1.text_area('enter your text here')
-col2.text('f"{ group_dict['label'] }: { round(p['score'] * 100, 1)}%"')
+with col1: 
+    input_text = text_area('enter your text here')
 
-st.write("Example: To promote gender diversity, ")
+with col2: 
+    st.write('f"{ group_dict['label'] }: { round(p['score'] * 100, 1)}%"')
+
+st.write("Example: To promote gender diversity, several new policies are being implemented")