zeroshot commited on
Commit
40f09bf
·
1 Parent(s): 2b6da15

edit handler

Browse files
Files changed (1) hide show
  1. handler.py +5 -1
handler.py CHANGED
@@ -20,4 +20,8 @@ class EndpointHandler:
20
  end = perf_counter()
21
  delta = end - start
22
 
23
- return prediction.json(), "latency: " + str(delta) + " secs."
 
 
 
 
 
20
  end = perf_counter()
21
  delta = end - start
22
 
23
+ return {
24
+ "labels":prediction.labels,
25
+ "scores": prediction.scores,
26
+ "latency (secs.)": str(delta)
27
+ }