Syeda-Sana-Bibi786 commited on
Commit
23c605d
·
verified ·
1 Parent(s): 162b49f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ def check_image(image):
9
  results = model(image)
10
  label = results[0]['label']
11
  score = results[0]['score']
12
- return f"Prediction: {label} (Confidence: {score*100:.2f}%)"
13
 
14
  # Gradio interface - EVERYTHING stays inside this Interface(...)
15
  iface = gr.Interface(
 
9
  results = model(image)
10
  label = results[0]['label']
11
  score = results[0]['score']
12
+ return f"Prediction: {label.capitalize()} (Confidence: {score*100:.2f}%)"
13
 
14
  # Gradio interface - EVERYTHING stays inside this Interface(...)
15
  iface = gr.Interface(