Update app.py
Browse files
app.py
CHANGED
|
@@ -143,7 +143,7 @@ def generate_performance_metrics():
|
|
| 143 |
|
| 144 |
with torch.no_grad():
|
| 145 |
for batch in val_loader:
|
| 146 |
-
inputs = tokenizer(
|
| 147 |
inputs = {key: val.to(device) for key, val in inputs.items()}
|
| 148 |
|
| 149 |
outputs = model(**inputs)
|
|
|
|
| 143 |
|
| 144 |
with torch.no_grad():
|
| 145 |
for batch in val_loader:
|
| 146 |
+
inputs = tokenizer(email_text, padding=True, truncation=True, max_length=256, return_tensors="pt")
|
| 147 |
inputs = {key: val.to(device) for key, val in inputs.items()}
|
| 148 |
|
| 149 |
outputs = model(**inputs)
|