Update app.py

app.py

@@ -128,6 +128,11 @@ text = "Bitcoin to the moon"
 
 encoding = tokenizer(text, return_tensors="pt")
 
+
+logits = outputs.logits
+logits.shape
+
+
 # apply sigmoid + threshold
 sigmoid = torch.nn.Sigmoid()
 probs = sigmoid(logits.squeeze().cpu())