Spaces:

ilyshi
/

shad_ml2_2

Runtime error

ilyshi commited on Mar 24, 2022

Commit

1044b9f

•

1 Parent(s): 50ac7e6

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ def make_prediction(text):
   tokenized_text = tokenizer(text, return_tensors='pt')
   with torch.no_grad():
     pred_logits = model(**tokenized_text).logits
-  st.markdown("### Category probability:")
   print_probs(pred_logits[0])

   tokenized_text = tokenizer(text, return_tensors='pt')
   with torch.no_grad():
     pred_logits = model(**tokenized_text).logits
+  st.markdown("### Category prediction:")
   print_probs(pred_logits[0])