Update README.md

README.md

@@ -41,12 +41,13 @@ license: mit
 ## Model Usage
 ```python
 from torchvision.transforms.functional import to_tensor
-from model import MicronFlow  # Assuming model implementation is available
+
+# load the model from the .pth file
 
 model = MicronFlow().eval()
-frame1 = to_tensor(image1).unsqueeze(0)  # Convert images to tensors
+frame1 = to_tensor(image1).unsqueeze(0)
 frame2 = to_tensor(image2).unsqueeze(0)
-flow = model(frame1, frame2)  # Optical flow prediction
+flow = model(frame1, frame2)
 ```
 
 ## License