Update app.py

app.py

@@ -3,20 +3,9 @@ import requests
 import numpy as np
 import torch
 import torch.nn as nn
-import torchvision.transforms as transforms
-# from badnet_m import BadNet
-
-import timm
-
-# model = timm.create_model("hf_hub:nateraw/resnet18-random", pretrained=True)
-# model.train()
-
-# model = BadNet(3, 10)
-# pipeline = pipeline.to('cuda:0')
-
+import torchvision.transforms as transform
 
 import os 
-
 import logging
 
 from transformers import WEIGHTS_NAME,AdamW,AlbertConfig,AlbertTokenizer,BertConfig,BertTokenizer
@@ -28,9 +17,6 @@ from transformers import glue_processors as processors
 import datasets
 from whitebox_utils.classifier import MyClassifier   
 
-import ssl
-ssl._create_default_https_context = ssl._create_unverified_context
-
 import random
 import numpy as np
 import torch