Update app.py

app.py

@@ -136,7 +136,7 @@ def train(mod, training_loader):
         token_type_ids = data['token_type_ids'].to(device, dtype=torch.long)
         targets = data['targets'].to(device, dtype=torch.float)
 
-        outs = model(input_ids, attention_mask, token_type_ids)
+        outs = mod(input_ids, attention_mask, token_type_ids)
 
         opt.zero_grad()
         loss = lossFN(outs, targets)