Update README.md
Browse files
README.md
CHANGED
|
@@ -69,7 +69,6 @@ model = SegformerForSemanticSegmentation.from_pretrained("pamixsun/segformer_for
|
|
| 69 |
inputs = processor(image, return_tensors="pt")
|
| 70 |
|
| 71 |
with torch.no_grad():
|
| 72 |
-
inputs.to(self.device)
|
| 73 |
outputs = self.seg_model(**inputs)
|
| 74 |
logits = outputs.logits.cpu()
|
| 75 |
|
|
|
|
| 69 |
inputs = processor(image, return_tensors="pt")
|
| 70 |
|
| 71 |
with torch.no_grad():
|
|
|
|
| 72 |
outputs = self.seg_model(**inputs)
|
| 73 |
logits = outputs.logits.cpu()
|
| 74 |
|