Update app.py

app.py

@@ -12,21 +12,17 @@ def prettier(results):
         location = [round(value, 2) for value in item['box'].values()]
         print(f'Detected {label} with confidence {score} at location {location}')
 
+# Function to process uploaded image and prepare input for model
 
 def input_image_setup(uploaded_file):
     if uploaded_file is not None:
-        #read the file into byte
-        bytes_data = uploaded_file.getvalue()        
-        image_parts=[
-            {
-                "mime_type": uploaded_file.type,
-                "data": bytes_data
-            }
-        ]
-        return image_parts
+        bytes_data = uploaded_file.getvalue()
+        image = Image.open(io.BytesIO(bytes_data))  # Convert bytes data to PIL image
+        return image
     else:
         raise FileNotFoundError("No file uploaded")
 
+
 #Streamlit App
 st.set_page_config(page_title="Image Detection")
 st.header("Object Detection Application")
@@ -46,7 +42,7 @@ if submit:
     image_data=input_image_setup(uploaded_file)
     st.subheader("The response is..")
     #process with model
-    inputs = processor(images=image_data, return_tensors="pt")
+    inputs = processor(images=image, return_tensors="pt")
     outputs = model(**inputs)
 
     # model predicts bounding boxes and corresponding COCO classes