Update app.py

app.py

@@ -41,20 +41,23 @@ async def predict_image(img: UploadFile, response: Response):
         # Predict the class of the image
         arr = model.predict(image)
 
-        if arr[0][0]==1:
-            labels='Fresh Apples'
-        elif arr[0][1]==1:
-            labels='Fresh Banana'
-        elif arr[0][2]==1:
-            labels='Fresh Oranges'
-        elif arr[0][3]==1:
-            labels='Rotten Apples'
-        elif arr[0][4]==1:
-            labels='Rotten Banana'
-        elif arr[0][5]==1:
-            labels='Rotten Oranges'
-
-        return {"result":labels}
+        arr = model.predict(image, batch_size=10)
+
+        # Mengambil indeks kelas dengan nilai probabilitas tertinggi
+        predicted_class_index = np.argmax(arr)
+
+        # Daftar label yang sesuai dengan kelas
+        class_labels = [
+            'Fresh Apples', 'Fresh Banana', 'Fresh Cucumber',
+            'Fresh Oranges',
+            'Rotten Apples', 'Rotten Banana', 'Rotten Cucumber',
+            'Rotten Oranges'
+        ]
+
+# Menentukan label
+        predicted_label = class_labels[predicted_class_index]
+
+        return {"result":predicted_label}
     except Exception as e:
         traceback.print_exc()
         response.status_code = 500