test handler
Browse files- handler.py +6 -10
handler.py
CHANGED
@@ -13,15 +13,11 @@ class EndpointHandler:
|
|
13 |
Args:
|
14 |
data (:obj:): prediction input text
|
15 |
"""
|
16 |
-
inputs = data.pop("inputs", data)
|
17 |
|
18 |
-
start = perf_counter()
|
19 |
-
prediction = self.pipeline(inputs)
|
20 |
-
end = perf_counter()
|
21 |
-
latency = end - start
|
22 |
|
23 |
-
return {
|
24 |
-
"labels": prediction.labels,
|
25 |
-
"scores": prediction.scores,
|
26 |
-
"latency (secs.)": latency
|
27 |
-
}
|
|
|
13 |
Args:
|
14 |
data (:obj:): prediction input text
|
15 |
"""
|
16 |
+
# inputs = data.pop("inputs", data)
|
17 |
|
18 |
+
# start = perf_counter()
|
19 |
+
# prediction = self.pipeline(inputs)
|
20 |
+
# end = perf_counter()
|
21 |
+
# latency = end - start
|
22 |
|
23 |
+
return {"greeting": "hi"}
|
|
|
|
|
|
|
|