Upload 3 files

README (1).md

@@ -0,0 +1,27 @@
+# ResNet18 Flower Classifier
+This model classifies images into one of five flower types.
+## Usage
+```python
+from torchvision import transforms
+from PIL import Image
+import torch
+from torchvision.models import resnet18
+
+model = resnet18(weights=None)
+model.load_state_dict(torch.load('path_to_model/pytorch_model.bin'))
+model.eval()
+
+transform = transforms.Compose([
+    transforms.Resize((224, 224)),
+    transforms.ToTensor(),
+    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
+])
+
+image = Image.open('path_to_image.jpg')
+image = transform(image).unsqueeze(0)
+
+with torch.no_grad():
+    output = model(image)
+    _, predicted = torch.max(output.data, 1)
+    print(predicted.item())
+```

config.json

@@ -0,0 +1 @@
+{"architecture": "resnet18", "num_labels": 5}

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:716e1d839aba5a49d9d1a95aa8e5b08de1c13bb71914da71a70763aa513b17c6
+size 44795890