Create README.md

README.md

@@ -0,0 +1,26 @@
+import torch
+from torchvision import transforms
+from PIL import Image
+
+# Load model
+model = torch.load("model.pth", map_location="cpu")
+model.eval()
+
+# Image preprocessing
+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])
+])
+
+# Load image
+image = Image.open("test_image.jpg").convert("RGB")
+input_tensor = transform(image).unsqueeze(0)
+
+# Prediction
+with torch.no_grad():
+    outputs = model(input_tensor)
+    predicted_class = torch.argmax(outputs, dim=1)
+
+print("Predicted class:", predicted_class.item())