Spaces:
Sleeping
Sleeping
Commit ·
2a1882c
1
Parent(s): c8d76fb
fix issue
Browse files- 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}
|