Update app.py

app.py

@@ -10,7 +10,7 @@ model = YOLOv10("doclayout_yolo_docstructbench_imgsz1024.pt")
 
 # Automatically select device
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
-st.write(f"Using device: {device}")
+#st.write(f"Using device: {device}")
 
 # Streamlit UI
 st.title("Document Layout Detection")
@@ -20,7 +20,7 @@ uploaded_file = st.file_uploader("Upload Image", type=["jpg", "jpeg", "png"])
 
 if uploaded_file is not None:
     # Display the uploaded image
-    st.image(uploaded_file, caption="Uploaded Image", use_container_width=True)
+    #st.image(uploaded_file, caption="Uploaded Image", use_container_width=True)
 
     # Load the uploaded image
     image = Image.open(uploaded_file).convert("RGB")
@@ -45,3 +45,8 @@ if uploaded_file is not None:
         # Display the annotated image
         st.image(annotated_image, caption="Annotated Image", use_container_width=True)
         st.success("Detection completed!")
+
+
+st.write("Application Created By Shubham Mhaske")
+st.write("Do have a look on Papers 📄 : - https://arxiv.org/pdf/2410.12628")
+st.write("Thanks to https://github.com/opendatalab")