Update README.md

README.md

@@ -1,20 +1,19 @@
-<h1>Jellyfish Classifier</h1>
+<h1>Flower Classifier</h1>
 finetuned from PyTorch ResNet18
 
-<h2>Jellyfish Label</h2>
-0: Compass Jellyfish<br>
-1: Blue Jellyfish<br>
-2: Mauve Stinger Jellyfish<br>
-3: Moon Jellyfish<br>
-4: Barrel Jellyfish<br>
-5: Lions Mane Jellyfish<br>
+<h2>Flower Label</h2>
+0: Lotus<br>
+1: Orchid<br>
+2: Tulip<br>
+3: Sunflower<br>
+4: Lilly<br>
 
 <h2>How to Use:</h2>
 <pre><code class="python">
 import torchvision
 from torchvision.models import resnet18
 model = resnet18(weights=None)
-model_weights_path = 'resnet18_jellyfish_classification.pth'
+model_weights_path = 'resnet18_flower_classification.pth'
 model.load_state_dict(torch.load(model_weights_path))
 model.eval()
 </pre>