Update README.md

README.md

@@ -5,6 +5,7 @@ language:
 ---
 
 2-step inference:
+
 ```python
 from PIL import Image
 import torch
@@ -41,8 +42,9 @@ class ImageBinaryClassifier(nn.Module):
         return self.layers(x)
 
 saved_model = ImageBinaryClassifier()
-saved_model.load_state_dict(torch.load(MODEL_PATH, map_location=torch.device('cpu')))
+saved_model.load_state_dict(torch.load("model.pt", map_location=torch.device('cpu')))
 saved_model = torch.compile(saved_model)
 saved_model.eval()
 
-prob = float(torch.sigmoid(saved_model(memb)))
+prob = float(torch.sigmoid(saved_model(memb)))
+```