Update app.py

app.py

@@ -1,3 +1,10 @@
+import gradio as gr
+import torch
+from torchvision.transforms import ToTensor
+from torchvision.models import resnet50
+from PIL import Image
+import torch.nn as nn
+
 # Load your PyTorch model
 model = resnet50(pretrained=False)
 model.fc = nn.Linear(model.fc.in_features, 2)