Update app.py

app.py

@@ -22,6 +22,10 @@ import matplotlib.pyplot as plt
 import seaborn as sns
 from albumentations.pytorch.transforms import ToTensorV2
 
+model = models.resnet50(pretrained=False)
+model.fc = nn.Linear(2048, num_classes)
+model.load_state_dict(torch.load('resnet_best.pth'), strict=True)
+
 st.title("some big ML function")
 
 uploaded_file = st.file_uploader("Choose a file")