Update app.py

app.py

@@ -27,11 +27,6 @@ import tensorflow as tf
 import gradio as gr
 
 # Your code using these packages can follow here
-
-
-# Your code using these packages can follow here
-
-
 def remove_background_deeplab(image):
     # Load the TensorFlow Lite model
     interpreter = tf.lite.Interpreter(model_path="2.tflite")
@@ -41,7 +36,7 @@ def remove_background_deeplab(image):
     output_details = interpreter.get_output_details()
 
     # Convert the Gradio Image object to a NumPy array
-    image_np = np.array(image)
+    image_np = np.array(image.data)
 
     # Resize the image to match the expected input size of the model
     input_size = (257, 257)
@@ -71,7 +66,9 @@ def remove_background_deeplab(image):
 
     return result_image
 
-
 # Example usage
-# remove_background_deeplab('images (2).jpeg')
-gr.Interface(fn=remove_background_deeplab,inputs=gr.Image(label='Drop an Image or Open Camera to Classify'),outputs=gr.Image()).launch()
+gr.Interface(
+    fn=remove_background_deeplab,
+    inputs=gr.Image(label='Drop an Image or Open Camera to Classify'),
+    outputs=gr.Image()
+).launch()