Echos updated

app.py

@@ -29,7 +29,7 @@ def train_model(start, end):
         )
         
         optimizer = torch.optim.Adam(model.parameters(), lr=5e-5)
-        for epoch in range(2):
+        for epoch in range(3):
             optimizer.zero_grad()
             outputs = model(**inputs, labels=inputs['input_ids'])
             loss = outputs.loss