Commit
•
cc27abb
1
Parent(s):
cb9bb26
Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -42,5 +42,5 @@ class EndpointHandler():
|
|
42 |
|
43 |
loss = outputs.loss
|
44 |
logits = outputs.logits
|
45 |
-
return {"logits": logits}
|
46 |
|
|
|
42 |
|
43 |
loss = outputs.loss
|
44 |
logits = outputs.logits
|
45 |
+
return {"logits": logits.tolist()}
|
46 |
|