Update app.py

app.py

@@ -37,6 +37,10 @@ group_dict = {
     28: 'Small-scale farmers',
     29: 'Other'}
 
+def predict(text):
+    preds = model([text])[0].item()
+    return group_dict[preds]
+
 # App 
 st.title("Identify references to vulnerable groups.")
 st.write("This app allows you to identify whether a text contains any references to vulnerable groups. This can, for example, be used to analyse policy documents.")
@@ -44,11 +48,11 @@ st.write("This app allows you to identify whether a text contains any references
 col1, col2 = st.columns(2)
 
 with col1: 
-    input_text = st.text_area('enter your text here')
+    input_text = st.text_area('Please enter your text here')
 
-preds = model(["input_text"])
+#preds = model(["input_text"])
 
 with col2: 
-    st.text(preds)
+    st.text(predict(input_text))
 
 st.write("Example: To promote gender diversity, several new policies are being implemented")