Alex Martin commited on
Commit
5a9eb5a
1 Parent(s): 89e215b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ def get_cats(text):
25
  prob = torch.sigmoid(outputs.logits).detach().numpy()[0]
26
 
27
  data = create_dict(prob, text)
28
- res = pd.DataFrame(data)
29
  st.table(res)
30
 
31
  words = "Take that, you funking cat-dragon! You smell really bad!"
 
25
  prob = torch.sigmoid(outputs.logits).detach().numpy()[0]
26
 
27
  data = create_dict(prob, text)
28
+ res = pd.DataFrame([data])
29
  st.table(res)
30
 
31
  words = "Take that, you funking cat-dragon! You smell really bad!"