mohamedahraf273 commited on
Commit
2a1882c
·
1 Parent(s): c8d76fb

fix issue

Browse files
Files changed (1) hide show
  1. api.py → app.py +1 -1
api.py → app.py RENAMED
@@ -16,5 +16,5 @@ with open("label_encoder.pkl", "rb") as f:
16
  def predict(embedding: np.ndarray):
17
  embedding = embedding.reshape(1, -1)
18
  pred = model.predict(embedding)
19
- pred_class = le.inverse_transform(pred)[0``]
20
  return {"result": pred_class}
 
16
  def predict(embedding: np.ndarray):
17
  embedding = embedding.reshape(1, -1)
18
  pred = model.predict(embedding)
19
+ pred_class = le.inverse_transform(pred)[0]
20
  return {"result": pred_class}