Mengubah takaran prediksi menjadi >90

main.py

@@ -88,9 +88,14 @@ async def predict(file: UploadFile = File(...)):
 
     preds = model.predict(img_array, verbose=0)
     index = np.argmax(preds[0])
-    predicted_label = CLASS_NAMES[index]
+
     confidence = round(float(np.max(preds[0])) * 100, 2)
 
+    if confidence >= 90.0:
+        predicted_label = CLASS_NAMES[index]
+    else:
+        predicted_label = "Tidak diketahui"
+
     probability = {}
     for i, prob in enumerate(preds[0]):
         probability[CLASS_NAMES[i]] = round(float(prob) * 100, 2)