Update app.py

app.py

@@ -10,7 +10,7 @@ st.title("Image Colorization App")
 st.write("Upload a black-and-white image, and this app will colorize it.")
 
 # Load the model
-colorization_system = ImageColorizationSystem("C:\\mlops project\\image-colorization-mlops\\artifacts\\trained_model\\cwgan_generator_final.pt", "C:\\mlops project\\image-colorization-mlops\\artifacts\\trained_model\\cwgan_critic_final.pt")
+colorization_system = ImageColorizationSystem("cwgan_generator_final.pt", "cwgan_critic_final.pt")
 
 # Upload image
 uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])