Update api.py

api.py

@@ -78,23 +78,23 @@ def detect_hands(image):
 
     return subImgs
 
-# def predict(image):
-#     img = image.astype("float32") / 255.0
-#     img = np.expand_dims(img, axis=0)
-#     pred = model.predict(img, verbose=0)
-#     label = label_encoder.inverse_transform([np.argmax(pred)])[0]
-#     return label
+def predict(image):
+    img = image.astype("float32") / 255.0
+    img = np.expand_dims(img, axis=0)
+    pred = model.predict(img, verbose=0)
+    label = label_encoder.inverse_transform([np.argmax(pred)])[0]
+    return label
 
 # Prediction function
-def predict(image):
-    img = cv2.resize(image, (64, 64))
-    img = np.expand_dims(img, axis=0)  # Add batch dimension
-    prediction = model.predict(img, verbose=0)
-    predicted_class = np.argmax(prediction)
-    print(predicted_class)
-    predicted_label = label_encoder.classes_[predicted_class]
-    print(predicted_label)
-    return predicted_label
+# def predict(image):
+#     img = cv2.resize(image, (64, 64))
+#     img = np.expand_dims(img, axis=0)  # Add batch dimension
+#     prediction = model.predict(img, verbose=0)
+#     predicted_class = np.argmax(prediction)
+#     print(predicted_class)
+#     predicted_label = label_encoder.classes_[predicted_class]
+#     print(predicted_label)
+#     return predicted_label
 
 @app.post("/ipredict")
 async def ipredict(image: UploadFile = File(...)):