Update app.py

app.py

@@ -1,3 +1,10 @@
+import torch
+from torchvision import transforms, datasets
+from torchvision.models import efficientnet_b0, EfficientNet_B0_Weights
+from torchvision.transforms.functional import InterpolationMode
+from PIL import Image
+import gradio as gr
+
 # Define el modelo y carga los pesos guardados
 model = efficientnet_b0(weights=EfficientNet_B0_Weights.DEFAULT)
 model.classifier[1] = torch.nn.Linear(in_features=1280, out_features=101)