Aye10032 commited on
Commit
db4f564
·
1 Parent(s): 2f2b372

change output type

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. top5_error_rate.py +2 -2
app.py CHANGED
@@ -21,7 +21,7 @@ local_path = Path(sys.path[0])
21
 
22
  default_value = pl.DataFrame({
23
  'predictions': ['0.82,0.95,0.6,0.14,0.15,0.70', '0.67,0.31,0.01,0.60,0.44,0.51', '0.57,0.06,0.69,0.07,0.96,0.72'],
24
- 'references': ['1', '3', '4']
25
  })
26
 
27
  iface = gr.Interface(
 
21
 
22
  default_value = pl.DataFrame({
23
  'predictions': ['0.82,0.95,0.6,0.14,0.15,0.70', '0.67,0.31,0.01,0.60,0.44,0.51', '0.57,0.06,0.69,0.07,0.96,0.72'],
24
+ 'references': ['1', '3', '1']
25
  })
26
 
27
  iface = gr.Interface(
top5_error_rate.py CHANGED
@@ -77,6 +77,6 @@ class Top5ErrorRate(evaluate.Metric):
77
  top5_error_rate = 1 - correct.mean()
78
 
79
  return {
80
- "accuracy": acc,
81
- "top5_error_rate": top5_error_rate
82
  }
 
77
  top5_error_rate = 1 - correct.mean()
78
 
79
  return {
80
+ "accuracy": float(acc),
81
+ "top5_error_rate": float(top5_error_rate)
82
  }