Update app.py

app.py

@@ -13,14 +13,8 @@ import torch.optim as optim
 from torch.utils.data import DataLoader, Dataset, DistributedSampler
 from transformers import AutoModel, AutoTokenizer
 from torchvision import models, transforms
-from sklearn.model_selection import train_test_split
-from sklearn.utils.class_weight import compute_class_weight
-from sklearn.metrics import precision_recall_fscore_support, accuracy_score
 from torch.cuda.amp import GradScaler, autocast
 import numpy as np
-import torch.multiprocessing as mp
-import torch.distributed as dist
-import matplotlib.pyplot as plt
 
 # --- Define the Model ---
 class FineGrainedClassifier(nn.Module):