update readme

README.md

@@ -2,4 +2,21 @@
 datasets:
 - garythung/trashnet
 pipeline_tag: image-classification
----
+---
+
+to load this state model use this step:
+#define the model
+model = resnet50(weights=ResNet50_Weights.IMAGENET1K_V2)
+for param in model_resnet.parameters():
+        param.requires_grad = False
+
+num_ftrs = model_resnet.fc.in_features
+model_resnet.fc = nn.Linear(num_ftrs, 6)
+
+# Load the weights
+state_dict = torch.load('trashnet_resnet50.pth')
+model.load_state_dict(state_dict)
+
+# Switch to evaluation mode
+model.eval()
+