Update app.py

app.py

@@ -25,7 +25,8 @@ def get_cats(text):
   prob = torch.sigmoid(outputs.logits).detach().numpy()[0]
   
   data = create_dict(prob, text)
-  st.table(data)
+  res = pd.Dataframe(data)
+  st.table(res)
 
 words = "Take that, you funking cat-dragon! You smell really bad!"
 text = st.text_area("Insert text for analysis below.", words)