Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -61,4 +61,4 @@ class EndpointHandler():
|
|
| 61 |
result = [{"label": label, "score": score.item()} for label, score in zip(self.categories, similarity)]
|
| 62 |
result = sorted(result, key=lambda x: x["score"], reverse=True)
|
| 63 |
|
| 64 |
-
return result[:
|
|
|
|
| 61 |
result = [{"label": label, "score": score.item()} for label, score in zip(self.categories, similarity)]
|
| 62 |
result = sorted(result, key=lambda x: x["score"], reverse=True)
|
| 63 |
|
| 64 |
+
return result[:50] # Return top 50
|