Update app.py

Mubazir-Bangkit-2023 · Dec 23, 2023 · dd6902a · dd6902a
1 parent ab5a621
commit dd6902a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/app.py b/app.py
@@ -39,7 +39,7 @@ async def predict_image(img: UploadFile, response: Response):
         image = np.expand_dims(image, axis=0)
 
         # Predict the class of the image
-        arr = model.predict(image, batch_size=10)
+        arr = model.predict(image)
 
         # Mengambil indeks kelas dengan nilai probabilitas tertinggi
         predicted_class_index = np.argmax(arr)