Spaces:

TeresaK
/

cpv_test

Runtime error

leavoigt commited on Jul 18, 2023

Commit

18ca863

1 Parent(s): 77a6d9d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -40,7 +40,7 @@ group_dict = {
 # Define prediction function
 def predict(text):
     preds = model([text])[0].item()
-  #  return group_dict[preds]
 # App
 st.title("Identify references to vulnerable groups.")

 # Define prediction function
 def predict(text):
     preds = model([text])[0].item()
+    return group_dict[preds]
 # App
 st.title("Identify references to vulnerable groups.")