Update app.py

app.py

@@ -8,7 +8,9 @@ import torch.nn as nn
 import cv2
 from albumentations.pytorch.transforms import ToTensorV2
 
-model = models.resnet50(pretrained=False)
+#Load model torch.load.state_dict(PATH)
+model = torch.load.state_dict('resnet_best.pth')
+#model = models.resnet50(pretrained=False)
 model.fc = nn.Linear(2048, 21)
 model.load_state_dict(torch.load('resnet_best.pth'), strict=True)#load weights from training run
 model.eval()
@@ -34,6 +36,4 @@ if uploaded_file is not None:
   elif ".csv" in uploaded_file.name: #csv check
     dataframe = pd.read_csv(uploaded_file)
     st.write(dataframe)
- 
-#Load model torch.load.state_dict(PATH)
-torch.load.state_dict('resnet_best.pth')
+