Update app.py

app.py

@@ -1,5 +1,16 @@
-import numpy as np
-
+import torch # ; print('Using torch version -', torch.__version__)
+if torch.cuda.is_available():
+    print('CUDA is available')
+    device = 'cuda'
+else:
+    print('CUDA NOT available')
+    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 gradio as gr
 
 def getoutput(input_img):