Update README.md

README.md

@@ -11,7 +11,7 @@ license: cc-by-nc-4.0
 # Model card for BVRA/resnet34.ft_in1k_df20m_299
 
 ## Model Details
-- **Model Type:** resnet34.ft_in1k_df20m_299
+- **Model Type:** Danish Fungi Classification 
 - **Model Stats:**
   - Params (M): ??
   - Image size: 299 x 299
@@ -29,12 +29,12 @@ from PIL import Image
 from urllib.request import urlopen
 model = timm.create_model("hf-hub:BVRA/resnet34.ft_in1k_df20m_299", pretrained=True)
 model = model.eval()
-train_transforms = T.Compose([T.Resize(299), 
+train_transforms = T.Compose([T.Resize((299, 299)), 
                               T.ToTensor(), 
-                              T.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])]) 
+                              T.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])]) 
 img = Image.open(PATH_TO_YOUR_IMAGE)
 output = model(train_transforms(img).unsqueeze(0))  # output is (batch_size, num_features) shaped tensor
-# output is a (1, num_features) shaped tensor
+
 ```
 
 ## Citation