Update app.py
Browse files
app.py
CHANGED
|
@@ -36,7 +36,8 @@ def classify(prompt: str):
|
|
| 36 |
truncation=True,
|
| 37 |
padding=True,
|
| 38 |
max_length=512
|
| 39 |
-
)
|
|
|
|
| 40 |
with torch.no_grad():
|
| 41 |
logits = model(**inputs).logits
|
| 42 |
probs = torch.softmax(logits, dim=-1).squeeze().cpu()
|
|
|
|
| 36 |
truncation=True,
|
| 37 |
padding=True,
|
| 38 |
max_length=512
|
| 39 |
+
)
|
| 40 |
+
|
| 41 |
with torch.no_grad():
|
| 42 |
logits = model(**inputs).logits
|
| 43 |
probs = torch.softmax(logits, dim=-1).squeeze().cpu()
|