Update README.md

README.md

@@ -55,6 +55,7 @@ import cv2
 import torch
 import numpy as np
 
+from torch import nn
 from transformers import AutoImageProcessor, SegformerForSemanticSegmentation
 
 image = cv2.imread('./example.jpg')
@@ -66,7 +67,7 @@ model = SegformerForSemanticSegmentation.from_pretrained("pamixsun/segformer_for
 inputs = processor(image, return_tensors="pt")
 
 with torch.no_grad():
-    outputs = self.seg_model(**inputs)
+    outputs = model(**inputs)
     logits = outputs.logits.cpu()
 
 upsampled_logits = nn.functional.interpolate(