Update utils/model.py

utils/model.py

@@ -73,11 +73,7 @@ def train_model(epochs, model_path):
         current_loss = losses.get("ner", float('inf'))
         epoch_losses.append(current_loss)
         
-        print(f"Losses at epoch {epoch + 1}: {losses}")
-        
-        # Stop training if the loss is zero
-        if current_loss == 0:
-            break
+        print(f"Losses at epoch {epoch + 1}: {losses}")        
         
         # Save the best model
         if current_loss < best_loss: