Create app.py

app.py

@@ -0,0 +1,21 @@
+import streamlit as st
+from transformers import DetrForObjectDetection, DetrFeatureExtractor
+from PIL import Image
+
+# Load model and feature extractor
+model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50")
+feature_extractor = DetrFeatureExtractor.from_pretrained("facebook/detr-resnet-50")
+
+st.title("Context-Aware Object Detection")
+
+# Upload an image
+uploaded_file = st.file_uploader("Choose an image...", type="jpg")
+if uploaded_file is not None:
+    image = Image.open(uploaded_file)
+    st.image(image, caption="Uploaded Image", use_column_width=True)
+
+    # Run detection
+    inputs = feature_extractor(images=image, return_tensors="pt")
+    outputs = model(**inputs)
+    # Display results here
+    st.write(outputs)