Upload folder using huggingface_hub

README.md

@@ -71,15 +71,17 @@ with torch.no_grad():
     image_tensor = torchvision.transforms.functional.to_tensor(image_array_resized).to(DEVICE_NAME)
     prediction = MODEL(image_tensor.unsqueeze(0)).squeeze(0).cpu().numpy()
     for i in (-prediction).argsort():
-        print(f'{classes[i]:>10}: {100 * prediction[i]:.1f}%')
+        print(f'{CLASSES[i]:>10}: {100 * prediction[i]:.1f}%')
 ```
 
 Output:
 ```
-     plant: 96.7%
-     human: 0.1%
-    insect: 0.1%
-  mushroom: 0.0%
+          plant: 91.3%
+unhealthy_plant: 53.3%
+           crop: 16.2%
+         insect: 0.4%
+          human: 0.1%
+       mushroom: 0.0%
 ```
 
 ###
@@ -114,13 +116,15 @@ INTERPRETER.invoke()
 logits = INTERPRETER.get_tensor(output_details[0]['index'])[0]
 prediction = tf.nn.sigmoid(logits).numpy()
 for i in (-prediction).argsort():
-    print(f'{classes[i]:>10}: {100 * prediction[i]:.1f}%')
+    print(f'{CLASSES[i]:>10}: {100 * prediction[i]:.1f}%')
 ```
 
 Output:
 ```
-     plant: 96.7%
-     human: 0.1%
-    insect: 0.1%
-  mushroom: 0.0%
+          plant: 91.3%
+unhealthy_plant: 53.3%
+           crop: 16.2%
+         insect: 0.4%
+          human: 0.1%
+       mushroom: 0.0%
 ```