Update app.py

app.py

@@ -29,17 +29,27 @@ detection_model = load_model(REPO_ID)
 student_id = "Student ID: 9053220B"
 
 # Create the Gradio interface
-interface = gr.Interface(
-    fn=predict,
-    inputs=[
-        gr.Image(type="pil", label="Input Image"),
-        gr.Slider(0, 1, value=0.5, label="Confidence Threshold"),  # Default to 0.5
-        gr.Slider(0, 1, value=0.6, label="IOU Threshold")  # Default to 0.6
-    ],
-    outputs=gr.Image(type="pil", label="Output Image"),
-    title="Object Detection with YOLOv8",
-    description=student_id
-)
+def create_interface():
+    # Persistent state for default values
+    confidence_default = gr.State(0.5)
+    iou_default = gr.State(0.6)
+
+    interface = gr.Interface(
+        fn=predict,
+        inputs=[
+            gr.Image(type="pil", label="Input Image"),
+            gr.Slider(0, 1, value=confidence_default.value, label="Confidence Threshold"),  # Default to 0.5
+            gr.Slider(0, 1, value=iou_default.value, label="IOU Threshold")  # Default to 0.6
+        ],
+        outputs=gr.Image(type="pil", label="Output Image"),
+        title="Object Detection with YOLOv8",
+        description=student_id,
+        live=False,
+    )
+
+    return interface
+
 
 # Launch the Gradio app
-interface.launch(share=True)
+app_interface = create_interface()
+app_interface.launch(share=True)