cleaning up code for the color analysis model x2

callaunchpad · Dec 7, 2024 · f80e801 · f80e801
1 parent 081327a
commit f80e801
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/color_analysis/color_analysis_model.py b/color_analysis/color_analysis_model.py
@@ -184,7 +184,6 @@ def main():
     print(f"Test Loss: {test_loss:.4f}, Test Accuracy: {test_accuracy:.2f}%")
 
     # Predict the season of the input image with the trained model
-    print(f"Predicting season for image: {args.image_path}")
     predicted_season = predict_image(model, args.image_path, class_labels, transform, device=device)
     if predicted_season:
         print(f"Predicted Season: {predicted_season}")