Update app.py

app.py

@@ -248,7 +248,7 @@ def predict(image):
     image = cv2.resize(image, (640, 640))
 
     # Convert the image to a numpy array and add a batch dimension
-    image = np.expand_dims(image, axis=0)
+    image = np.expand_dims(image, axis=0).astype(np.float32)
 
     # Perform inference
     output = model.run(None, {input_name: image})