Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ def predict(image):
|
|
22 |
|
23 |
# Return dictionary: class name -> confidence (no extra keys!)
|
24 |
labels = [model.config.id2label[i] for i in range(len(probs))]
|
25 |
-
return {labels[i]: float(f"{probs[i]:.
|
26 |
|
27 |
# Gradio Interface
|
28 |
def main():
|
|
|
22 |
|
23 |
# Return dictionary: class name -> confidence (no extra keys!)
|
24 |
labels = [model.config.id2label[i] for i in range(len(probs))]
|
25 |
+
return {labels[i]: float(f"{probs[i]:.8f}") for i in range(len(labels))}
|
26 |
|
27 |
# Gradio Interface
|
28 |
def main():
|