Update app.py

app.py

@@ -1,16 +1,6 @@
 from UNet import *
 from make_predictions import *
-import torch # ; print('Using torch version -', torch.__version__)
-if torch.cuda.is_available():
-    device = 'cuda'
-else:
-    device = 'cpu'
-from torch.nn import Module, Conv2d, ReLU, ModuleList, MaxPool2d, ConvTranspose2d, BCELoss, BCEWithLogitsLoss, functional as F
-from torch.optim import Adam
-from torchvision import transforms
-from torchvision.transforms import CenterCrop
-from torch.utils.data import Dataset, DataLoader
-import cv2
+import torch
 import gradio as gr
 
 def getoutput(input_img):