Spaces:

d4niel92
/

surface-classification

Sleeping

d4niel92 commited on Jun 1, 2023

Commit

960819e

1 Parent(s): 9db43e7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ class_labels = ['unlabeled', 'paved-area', 'dirt', 'grass', 'gravel', 'water', '
 def classify(im):
-    inputs = extractor(images=im, return_tensors="pt").
     outputs = model(**inputs)
     logits = outputs.logits
     classes = logits[0].detach().cpu().numpy().argmax(axis=0)

 def classify(im):
+    inputs = extractor(images=im, return_tensors="pt")
     outputs = model(**inputs)
     logits = outputs.logits
     classes = logits[0].detach().cpu().numpy().argmax(axis=0)