Update app.py

app.py

@@ -5,7 +5,7 @@ import pandas as pd
 import numpy as np
 import joblib
 from PIL import Image
-from torchvision import transforms
+from torchvision import transforms,models
 from sklearn.preprocessing import LabelEncoder
 from gradio import Interface, Image, Label
 from huggingface_hub import Repository