pamixsun commited on
Commit
df038eb
1 Parent(s): 3d72fc8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -1
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