Spaces:
Sleeping
Sleeping
Commit
·
678a706
1
Parent(s):
9abff41
Update app.py
Browse files
app.py
CHANGED
@@ -37,7 +37,7 @@ def inference(text):
|
|
37 |
tflite_pred = interpreter.get_tensor(output_details["index"])[0]
|
38 |
tflite_pred = softmax(tflite_pred)
|
39 |
tflite_pred_argmax = np.argmax(tflite_pred)
|
40 |
-
tflite_pred = f"{label_encoder.inverse_transform([tflite_pred_argmax])
|
41 |
return tflite_pred
|
42 |
|
43 |
|
|
|
37 |
tflite_pred = interpreter.get_tensor(output_details["index"])[0]
|
38 |
tflite_pred = softmax(tflite_pred)
|
39 |
tflite_pred_argmax = np.argmax(tflite_pred)
|
40 |
+
tflite_pred = f"{label_encoder.inverse_transform([tflite_pred_argmax])[0]}"
|
41 |
return tflite_pred
|
42 |
|
43 |
|