Update app.py

app.py

@@ -1,4 +1,20 @@
 import gradio as gr
+import torch
+import torch.nn as nn
+import torch.optim as optim
+import torchvision
+from torch.utils.data import Dataset
+from torchvision.io import read_image
+from torch.utils.data import DataLoader
+import torchvision.models as models
+import os
+import numpy as np
+import matplotlib.pyplot as plt
+import pandas as pd
+from sklearn.model_selection import GroupShuffleSplit
+from torchvision import transforms, utils
+import time
+import detectron2
 
 def greet(name):
     return "Hello " + name + "!!"