Update app.py

app.py

@@ -1,8 +1,13 @@
 
 from transformers import pipeline
 import streamlit as st
+# Load model directly
+from transformers import AutoImageProcessor, AutoModelForObjectDetection
+
 def model():
-    pipe = pipeline("object-detection", model="facebook/detr-resnet-50")
+    processor = AutoImageProcessor.from_pretrained("facebook/detr-resnet-50")
+    model = AutoModelForObjectDetection.from_pretrained("facebook/detr-resnet-50")
+    pipe = pipeline("object-detection", model="processor)
     output=pipe("dogs1.webp")    
     return output