DmitriySv commited on
Commit
cb22199
·
verified ·
1 Parent(s): 100eb7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -13,6 +13,8 @@ def classify(text):
13
  inputs = tokenizer(text, padding=True, truncation=True, max_length=512, return_tensors="pt").to(device)
14
 
15
  with torch.no_grad():
 
 
16
  logits_task1, logits_task2 = model(**inputs)
17
 
18
  pred_task1 = torch.argmax(logits_task1, dim=1).item()
 
13
  inputs = tokenizer(text, padding=True, truncation=True, max_length=512, return_tensors="pt").to(device)
14
 
15
  with torch.no_grad():
16
+ outputs = model(**inputs)
17
+ print(outputs)
18
  logits_task1, logits_task2 = model(**inputs)
19
 
20
  pred_task1 = torch.argmax(logits_task1, dim=1).item()