Update app.py

app.py

@@ -6,6 +6,8 @@ from PIL import Image
 # Load the trained model
 model_path = "cifar_net.pth"
 model = torch.load(model_path, map_location=torch.device('cpu'))
+model = YourModelClass()  # Replace YourModelClass with the appropriate model class
+model.load_state_dict(state_dict)
 model.eval()
 
 # Define class labels for CIFAR-10