Update app.py

app.py

@@ -1,4 +1,4 @@
-import streamlit as st
+#import streamlit as st
 from io import StringIO
 from PIL import Image
 import pandas as pd
@@ -9,18 +9,18 @@ import os
 import torch
 import torch.nn as nn
 from torch.utils.data import Dataset, DataLoader
-import torchvision.models as models
-from torchinfo import summary
+#import torchvision.models as models
+#from torchinfo import summary
 
-from sklearn.model_selection import StratifiedKFold
-from sklearn.metrics import accuracy_score
+#from sklearn.model_selection import StratifiedKFold
+#from sklearn.metrics import accuracy_score
 
 from tqdm import tqdm
 import cv2
-import albumentations as A # Albumentations is a computer vision tool that boosts the performance of deep convolutional neural networks. (https://albumentations.ai/)
+#import albumentations as A # Albumentations is a computer vision tool that boosts the performance of deep convolutional neural networks. (https://albumentations.ai/)
 import matplotlib.pyplot as plt
 import seaborn as sns
-from albumentations.pytorch.transforms import ToTensorV2
+#from albumentations.pytorch.transforms import ToTensorV2
 
 model = models.resnet50(pretrained=False)
 model.fc = nn.Linear(2048, num_classes)