Ateeqq commited on
Commit
c1cda26
·
verified ·
1 Parent(s): 4a7a631

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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]:.6f}") for i in range(len(labels))}
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():