Update app.py

app.py

@@ -68,14 +68,21 @@ def infer(model, in_pil_img):
     return output_pil_img
 
 
-with gr.Blocks(title="DETR Object Detection - ClassCat",
-                    css=".gradio-container {background:lightyellow;}"
-               ) as demo:
-    #sample_index = gr.State([])
-
-    gr.HTML("""<div style="font-family:'Times New Roman', 'Serif'; font-size:16pt; font-weight:bold; text-align:center; color:royalblue;">DETR Object Detection</div>""")
-
-    gr.HTML("""<h4 style="color:navy;">1. Select a model.</h4>""")
+with gr.Blocks(title="DETR Object Detection by orYx Models") as demo:
+    gr.HTML("""
+    <style>
+    .logo {
+        position: absolute;
+        top: 10px;
+        right: 10px;
+        width: 100px; /* Adjust the width of the logo as needed */
+        height: auto;
+    }
+    </style>
+    <div style="font-family:'Times New Roman', 'Serif'; font-size:16pt; font-weight:bold; text-align:center; color:royalblue;">DETR Object Detection</div>
+    <img class="logo" src="https://huggingface.co/spaces/orYx-models/object-detection-facebook-ResNets/blob/main/oryx_logo%20(2).png" alt="Logo">
+    <h4 style="color:navy;">1. Select a model.</h4>
+    """)
 
     model = gr.Radio(["detr-resnet-50", "detr-resnet-101"], value="detr-resnet-50", label="Model name")
 
@@ -87,7 +94,8 @@ with gr.Blocks(title="DETR Object Detection - ClassCat",
         input_image = gr.Image(label="Input image", type="pil")
         output_image = gr.Image(label="Output image with predicted instances", type="pil")
 
-    gr.Examples(['samples/cats.jpg', 'samples/detectron2.png', 'samples/cat.jpg', 'samples/hotdog.jpg'], inputs=input_image)
+    gr.Examples(['https://huggingface.co/spaces/orYx-models/object-detection-facebook-ResNets/blob/main/traffic.jpg',
+                 'https://huggingface.co/spaces/orYx-models/object-detection-facebook-ResNets/blob/main/flyover.jpg'], inputs=input_image)
 
     gr.HTML("""<br/>""")
     gr.HTML("""<h4 style="color:navy;">3. Then, click "Infer" button to predict object instances. It will take about 10 seconds (on cpu)</h4>""")