Add application file

app.py

@@ -397,7 +397,7 @@ epochs = 5
 
 train_dataloaderd, test_dataloaderd = DataLoader(subset_train, batch_size=BATCH_SIZE, shuffle=True), DataLoader(subset_test, batch_size=BATCH_SIZE, shuffle=False) # change data here IMPORTANT!!!!!!!!!!!!!!!!!!!!!!!
 
-# engine.train(modeld, train_dataloaderd, test_dataloaderd, optimizerd, loss_fn, epochs, device)
+# train(modeld, train_dataloaderd, test_dataloaderd, optimizerd, loss_fn, epochs, device)
 
 def pred_image(image_path: str, model: torch.nn.Module = modeld, class_names: List[str] = training_datab.classes, image_size: Tuple[int, int] = (224, 224), transform: torchvision.transforms = ConvNeXt_Base_Weights.DEFAULT.transforms(), device: torch.device = device):
     # Open image