BhumikaMak commited on
Commit
408a665
·
1 Parent(s): 084f1b5

Update: background template

Browse files
Files changed (1) hide show
  1. app.py +39 -40
app.py CHANGED
@@ -46,49 +46,48 @@ def process_image(sample_choice, uploaded_image, yolo_versions=["yolov5"]):
46
  result_images.append((Image.fromarray(image), f"{yolo_version} not yet implemented."))
47
  return result_images
48
 
49
- with gr.Blocks(css=".pink-bg { background-color: pink; } .orchid-green { color: #32CD32; font-weight: bold; }") as interface:
50
- with gr.Box(elem_classes="orchid-green-bg"):
51
- gr.Markdown("# XAI: Visualize Object Detection of Your Models")
52
- gr.Markdown("Select a sample image to visualize object detection.")
53
- default_sample = "Sample 1"
54
- with gr.Row():
55
- # Left side: Sample selection and upload image
56
- with gr.Column():
57
- sample_selection = gr.Radio(
58
- choices=list(sample_images.keys()),
59
- label="Select a Sample Image",
60
- type="value",
61
- value=default_sample, # Set default selection
62
- )
63
- # Upload image below sample selection
64
- gr.Markdown("**Or upload your own image:**")
65
- upload_image = gr.Image(
66
- label="Upload an Image",
67
- type="pil", # Correct type for file path compatibility
68
- )
69
- # Right side: Selected sample image display
70
- sample_display = gr.Image(
71
- value=load_sample_image(default_sample),
72
- label="Selected Sample Image",
73
  )
74
-
75
- sample_selection.change(
76
- fn=load_sample_image,
77
- inputs=sample_selection,
78
- outputs=sample_display,
 
 
 
 
 
79
  )
 
 
 
 
 
 
80
 
81
- selected_models = gr.CheckboxGroup(
82
- choices=["yolov5", "yolov8s"],
83
- value=["yolov5"],
84
- label="Select Model(s)",
85
- )
86
- result_gallery = gr.Gallery(label="Results", elem_id="gallery", rows=2, height=500)
87
 
88
- gr.Button("Run").click(
89
- fn=process_image,
90
- inputs=[sample_selection, upload_image, selected_models], # Include both options
91
- outputs=result_gallery,
92
- )
93
 
94
  interface.launch(share=True)
 
46
  result_images.append((Image.fromarray(image), f"{yolo_version} not yet implemented."))
47
  return result_images
48
 
49
+ with gr.Blocks(css=".orchid-green-bg { background-color: #32CD32 !important; padding: 20px; }") as interface:
50
+ gr.Markdown("# XAI: Visualize Object Detection of Your Models")
51
+ gr.Markdown("Select a sample image to visualize object detection.")
52
+ default_sample = "Sample 1"
53
+ with gr.Row(elem_classes="orchid-green-bg"):
54
+ # Left side: Sample selection and upload image
55
+ with gr.Column():
56
+ sample_selection = gr.Radio(
57
+ choices=list(sample_images.keys()),
58
+ label="Select a Sample Image",
59
+ type="value",
60
+ value=default_sample, # Set default selection
 
 
 
 
 
 
 
 
 
 
 
 
61
  )
62
+ # Upload image below sample selection
63
+ gr.Markdown("**Or upload your own image:**")
64
+ upload_image = gr.Image(
65
+ label="Upload an Image",
66
+ type="pil", # Correct type for file path compatibility
67
+ )
68
+ # Right side: Selected sample image display
69
+ sample_display = gr.Image(
70
+ value=load_sample_image(default_sample),
71
+ label="Selected Sample Image",
72
  )
73
+
74
+ sample_selection.change(
75
+ fn=load_sample_image,
76
+ inputs=sample_selection,
77
+ outputs=sample_display,
78
+ )
79
 
80
+ selected_models = gr.CheckboxGroup(
81
+ choices=["yolov5", "yolov8s"],
82
+ value=["yolov5"],
83
+ label="Select Model(s)",
84
+ )
85
+ result_gallery = gr.Gallery(label="Results", elem_id="gallery", rows=2, height=500)
86
 
87
+ gr.Button("Run").click(
88
+ fn=process_image,
89
+ inputs=[sample_selection, upload_image, selected_models], # Include both options
90
+ outputs=result_gallery,
91
+ )
92
 
93
  interface.launch(share=True)