Spaces:
Runtime error
Runtime error
Commit
·
167eb46
1
Parent(s):
5869983
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ def classify(im):
|
|
16 |
confidences = {label: float(probs[i]) for i, label in enumerate(labels)}
|
17 |
return confidences
|
18 |
|
19 |
-
|
20 |
|
21 |
def classify(im):
|
22 |
features = feature_extractor(im, return_tensors='pt')
|
|
|
16 |
confidences = {label: float(probs[i]) for i, label in enumerate(labels)}
|
17 |
return confidences
|
18 |
|
19 |
+
|
20 |
|
21 |
def classify(im):
|
22 |
features = feature_extractor(im, return_tensors='pt')
|