Spaces:
Runtime error
Runtime error
shyamgupta196
commited on
Commit
•
4dbe981
1
Parent(s):
41a49cf
app.py
CHANGED
|
@@ -10,7 +10,6 @@ def seg(image):
|
|
| 10 |
inputs = feature_extractor(images=image, return_tensors="pt")
|
| 11 |
outputs = model(**inputs)
|
| 12 |
logits = outputs.logits
|
| 13 |
-
# model predicts one of the 1000 ImageNet classes
|
| 14 |
predicted_class_idx = logits.argmax(-1).item()
|
| 15 |
return model.config.id2label[predicted_class_idx]
|
| 16 |
|
|
|
|
| 10 |
inputs = feature_extractor(images=image, return_tensors="pt")
|
| 11 |
outputs = model(**inputs)
|
| 12 |
logits = outputs.logits
|
|
|
|
| 13 |
predicted_class_idx = logits.argmax(-1).item()
|
| 14 |
return model.config.id2label[predicted_class_idx]
|
| 15 |
|