leygit commited on
Commit
c2df036
·
verified ·
1 Parent(s): 32062ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(email, 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)
 
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)