ayethuzar commited on
Commit
1047b7a
·
unverified ·
1 Parent(s): 0dc667c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ if submit:
22
  batch = tokenizer(test, padding = True, truncation = True, max_length = 512, return_tensors = "pt")
23
 
24
  with torch.no_grad():
25
- outputs = model(**batch, labels = torch.tensor([1, 0]))
26
  st.write(outputs)
27
  predictions = F.softmax(outputs.logits, dim = 1)
28
  st.write(predictions)
 
22
  batch = tokenizer(test, padding = True, truncation = True, max_length = 512, return_tensors = "pt")
23
 
24
  with torch.no_grad():
25
+ outputs = model(**batch)
26
  st.write(outputs)
27
  predictions = F.softmax(outputs.logits, dim = 1)
28
  st.write(predictions)