app file changes

app.py

@@ -5,7 +5,7 @@ from model_builder import TinyVGG  # Replace with your actual model import
 
 # Load the model
 model = TinyVGG(input_shape=3, hidden_units=10, output_shape=3)
-model.load_state_dict(torch.load("pytorch_model.bin", map_location=torch.device("cpu"),  weights_only=True))
+model.load_state_dict(torch.load("05_going_modular_script_mode_tinyvgg_model.pth", map_location=torch.device("cpu")))
 model.eval()
 
 # Define a preprocessing pipeline