sam-brause commited on
Commit ·
7c8801e
1
Parent(s): 6491258
make changes based on input from gemini
Browse files- handler.py +3 -1
handler.py
CHANGED
|
@@ -61,4 +61,6 @@ class EndpointHandler:
|
|
| 61 |
predictions = probabilities.squeeze().tolist()
|
| 62 |
output = [issue for issue, prob in zip(self.supported_issues, predictions) if prob > 0.5]
|
| 63 |
|
| 64 |
-
return {"predictions": output}
|
|
|
|
|
|
|
|
|
| 61 |
predictions = probabilities.squeeze().tolist()
|
| 62 |
output = [issue for issue, prob in zip(self.supported_issues, predictions) if prob > 0.5]
|
| 63 |
|
| 64 |
+
return {"predictions": output}
|
| 65 |
+
|
| 66 |
+
EndpointHandler = EndpointHandler
|