Spaces:

accelera-ai
/

open_mp_classifier

Sleeping

mohamedahraf273 commited on Feb 4

Commit

c8d76fb

1 Parent(s): 0234723

fix api

Files changed (1) hide show

api.py CHANGED Viewed

@@ -16,5 +16,5 @@ with open("label_encoder.pkl", "rb") as f:
 def predict(embedding: np.ndarray):
     embedding = embedding.reshape(1, -1)
     pred = model.predict(embedding)
-    pred_class = le.inverse_transform(pred)[0]
     return {"result": pred_class}

 def predict(embedding: np.ndarray):
     embedding = embedding.reshape(1, -1)
     pred = model.predict(embedding)
+    pred_class = le.inverse_transform(pred)[0``]
     return {"result": pred_class}