d4niel92 commited on
Commit
960819e
1 Parent(s): 9db43e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ class_labels = ['unlabeled', 'paved-area', 'dirt', 'grass', 'gravel', 'water', '
9
 
10
 
11
  def classify(im):
12
- inputs = extractor(images=im, return_tensors="pt").
13
  outputs = model(**inputs)
14
  logits = outputs.logits
15
  classes = logits[0].detach().cpu().numpy().argmax(axis=0)
 
9
 
10
 
11
  def classify(im):
12
+ inputs = extractor(images=im, return_tensors="pt")
13
  outputs = model(**inputs)
14
  logits = outputs.logits
15
  classes = logits[0].detach().cpu().numpy().argmax(axis=0)