leavoigt commited on
Commit
fb908cd
1 Parent(s): 291495c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -55,10 +55,10 @@ input_text = st.text_area('Please enter your text here')
55
  preds = model(input_text)
56
 
57
  # Select lab
58
- def get_label(prediction_tensor):
59
- print(prediction_tensor.index("1"))
60
  #key = prediction_tensor.index(1)
61
  #return group_dict[key]
62
 
63
-
64
- st.text(get_label(preds))
 
55
  preds = model(input_text)
56
 
57
  # Select lab
58
+ #def get_label(prediction_tensor):
59
+ # print(prediction_tensor.index("1"))
60
  #key = prediction_tensor.index(1)
61
  #return group_dict[key]
62
 
63
+ st.text(preds)
64
+ #st.text(get_label(preds))