alphamike commited on
Commit
cca0e97
·
1 Parent(s): 419fc11

Enhance README.md with inference instructions and Gradio UI details

Browse files

- Added sections for CLI inference commands for both baseline and GNN checkpoints.
- Included instructions for launching the Gradio app and using it to upload images and select models.
- Updated the README to provide clearer guidance on inference processes.

Files changed (1) hide show
  1. README.md +24 -0
README.md CHANGED
@@ -125,6 +125,30 @@ Important: the processing pipeline is strict chair-only (`primary_object_class:
125
 
126
  `python scripts/compare_baseline_vs_gnn.py --baseline-run smoke_baseline --gnn-run smoke_gnn`
127
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
128
  ## Expected Outputs
129
 
130
  Data artifacts:
 
125
 
126
  `python scripts/compare_baseline_vs_gnn.py --baseline-run smoke_baseline --gnn-run smoke_gnn`
127
 
128
+ ### 3) Inference and Gradio UI
129
+
130
+ #### CLI inference
131
+
132
+ - Baseline checkpoint:
133
+
134
+ `python scripts/infer.py --checkpoint outputs/smoke_baseline/best.pt --image path/to/chair.jpg --top-k 5 --threshold 0.5`
135
+
136
+ - GNN checkpoint:
137
+
138
+ `python scripts/infer.py --checkpoint outputs/smoke_gnn/best.pt --image path/to/chair.jpg --top-k 5 --threshold 0.5`
139
+
140
+ #### Gradio app (local)
141
+
142
+ - Launch:
143
+
144
+ `python app.py`
145
+
146
+ - Then:
147
+ - Open the printed URL.
148
+ - Upload a chair image.
149
+ - Choose **baseline** or **gnn**.
150
+ - Adjust threshold/top-k as needed.
151
+
152
  ## Expected Outputs
153
 
154
  Data artifacts: