Update app.py

app.py

@@ -38,9 +38,9 @@ group_dict = {
     29: 'Other'}
 
 # Define prediction function 
-def predict(text):
-    preds = model(text)
-    return group_dict[preds]
+#def predict(text):
+ #   preds = model(text)
+  #  return group_dict[preds]
 
 # App 
 st.title("Identify references to vulnerable groups.")
@@ -51,16 +51,11 @@ into national climate policies, governments can ensure equitable outcomes, promo
 fostering a more sustainable and inclusive society as we navigate the challenges posed by climate change.This app allows you to identify whether a text contains any 
 references to vulnerable groups, for example when talking about policy documents.""")
 
-#col1, col2 = st.columns(2)
 
 # Create text input box
 input_text = st.text_area(label='Please enter your text here', value="This policy has been implemented to support women.")
 
-st.write('Prediction:', predict(input_text))
-
-# Create the output box
-#output=""
-#st.text_area(label="Prediction:", height=350)
+st.write('Prediction:', model(input_text))
 
 # Make predictions
 #preds = model(input_text)