Update app.py

app.py

@@ -1,4 +1,31 @@
-model = MyLightningModule()
+import torch
+import torch.nn as nn
+import pytorch_lightning as pl
+from torchvision.datasets import MNIST
+from torchvision.transforms import ToTensor
+from torch.utils.data import DataLoader, random_split
+import torch
+import albumentations as A
+from albumentations.pytorch import ToTensorV2
+
+from torchvision import transforms
+import numpy as np
+import torch
+from torchvision import datasets
+from torch.utils.data import Dataset, DataLoader
+from torchvision.transforms import ToTensor
+from torchmetrics import Accuracy
+from torch.nn import functional as F
+import matplotlib.pyplot as plt
+
+model = Cifar10SearchDataset()
 trainer = pl.Trainer(accelerator="auto",max_epochs=24)
+
+trainer.fit(model)
+trainer.test(model)
+
+
+#model = MyLightningModule()
+#trainer = pl.Trainer(accelerator="auto",max_epochs=24)
 #trainer = pl.Trainer()
-trainer.fit(model, train_loader, test_loader)
+#trainer.fit(model, train_loader, test_loader)