Update app.py

app.py

@@ -22,7 +22,11 @@ if upload_option == "Upload Image":
     uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg"])
     if uploaded_file is not None:
         image = Image.open(uploaded_file)
-        results = model(image)  
+        # Convert PIL image to numpy array
+        image_np = np.array(image)
+        # Convert numpy array to PyTorch tensor
+        image_tensor = torch.from_numpy(image_np)
+        results = model(image_tensor)  
         st.image(results.render()[0], caption='Detected Objects', use_column_width=True) 
 
 # Real-Time Webcam Detection
@@ -45,7 +49,9 @@ if upload_option == "Real-Time Webcam":
 
             # Convert to RGB and detect objects
             frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-            results = model(frame_rgb)
+            # Convert numpy array to PyTorch tensor
+            frame_tensor = torch.from_numpy(frame_rgb)
+            results = model(frame_tensor)
 
             # Render and display results
             annotated_frame = results.render()[0]