Update app.py

app.py

@@ -1,8 +1,16 @@
 import os
 import gradio as gr
 from transformers import pipeline
+from transformers import DetrForSegmentation, DetrConfig
 
-od_pipe = pipeline("object-detection", "facebook/detr-resnet-50")
+# Initialize the configuration for DetrForObjectDetection
+config = DetrConfig.from_pretrained("facebook/detr-resnet-50")
+
+# Create the model for object detection using the specified configuration
+model = DetrForSegmentation.from_pretrained("facebook/detr-resnet-50", config=config)
+
+# Updated function call
+results = processed_image(model, image, size={'longest_edge': 800})
 
 def get_pipeline_prediction(pil_image):
     # first get the pipeline output given the pil image
@@ -10,6 +18,7 @@ def get_pipeline_prediction(pil_image):
     # Then Process the image using the pipeline output
     processed_image = render_results_in_image(pil_image,
                                             pipeline_output)
+   
     return processed_image