Hugging Face's logo

Spaces:
alkzar90
/
rock-glacier-segmentation
Running

App Files Community
alkzar90 commited on
Commit
9758099
•
1 Parent(s): cb54c63

fix typo

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ MODEL_PATH="./best_model_test/"
12
  device = torch.device("cpu")
13
 
14
  preprocessor = transforms.Compose([
15
- transforms.resize(128),
16
  transforms.ToTensor()
17
  ])
18
  model = SegformerForSemanticSegmentation.from_pretrained(MODEL_PATH)
 
12
  device = torch.device("cpu")
13
 
14
  preprocessor = transforms.Compose([
15
+ transforms.Resize(128),
16
  transforms.ToTensor()
17
  ])
18
  model = SegformerForSemanticSegmentation.from_pretrained(MODEL_PATH)