MrDdz commited on
Commit
11f0536
·
1 Parent(s): ab81402

updated app

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -72,7 +72,8 @@ tokenizer, model = get_model()
72
 
73
  if text:
74
  text_sample = tokenizer([text], padding = 'max_length')
75
- output = model(text_sample)
76
- st.write("Logits: ",output.logits)
77
- y_pred = np.argmax(output.logits.detach().numpy(),axis =1)
78
- st.write("Prediction :",d[y_pred[0]])
 
 
72
 
73
  if text:
74
  text_sample = tokenizer([text], padding = 'max_length')
75
+ print(text_sample)
76
+ # output = model(text_sample)
77
+ # st.write("Logits: ",output.logits)
78
+ # y_pred = np.argmax(output.logits.detach().numpy(),axis =1)
79
+ # st.write("Prediction :",d[y_pred[0]])