sam-brause commited on
Commit
7c8801e
·
1 Parent(s): 6491258

make changes based on input from gemini

Browse files
Files changed (1) hide show
  1. 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