MrDdz commited on
Commit
94dfa81
1 Parent(s): dfd7342
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -74,7 +74,7 @@ tokenizer, model = get_model()
74
  if text and button:
75
  text_sample = tokenizer([text], padding = 'max_length')
76
  print(text_sample)
77
- output = model(**text_sample)
78
  # st.write("Logits: ",output.logits)
79
  # y_pred = np.argmax(output.logits.detach().numpy(),axis =1)
80
  # st.write("Prediction :",d[y_pred[0]])
 
74
  if text and button:
75
  text_sample = tokenizer([text], padding = 'max_length')
76
  print(text_sample)
77
+ output = model(text_sample)
78
  # st.write("Logits: ",output.logits)
79
  # y_pred = np.argmax(output.logits.detach().numpy(),axis =1)
80
  # st.write("Prediction :",d[y_pred[0]])