mrdbourke commited on
Commit
32cf9f0
·
verified ·
1 Parent(s): 4fc06e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -157,14 +157,19 @@ demo = gr.Interface(
157
  ],
158
  outputs = gr.Image(type="pil", label="Detections"),
159
  title = "LLMDet Demo: Open-Vocabulary Grounded Object Detection",
160
- description = """Upload an image, enter text queries, and adjust thresholds to see detections.
161
 
162
  Adapted from LLMDet GitHub repo [Hugging Face demo](https://github.com/iSEE-Laboratory/LLMDet/tree/main/hf_model).
163
 
 
 
164
  See original:
165
  * [LLMDet GitHub](https://github.com/iSEE-Laboratory/LLMDet/tree/main?tab=readme-ov-file)
166
  * [LLMDet Paper](https://arxiv.org/abs/2501.18954) - LLMDet: Learning Strong Open-Vocabulary Object Detectors under the Supervision of Large Language Models
167
- * [LLMDet model checkpoint](https://huggingface.co/fushh7/llmdet_swin_large_hf)
 
 
 
168
  """
169
  )
170
 
 
157
  ],
158
  outputs = gr.Image(type="pil", label="Detections"),
159
  title = "LLMDet Demo: Open-Vocabulary Grounded Object Detection",
160
+ description = f"""Upload an image, enter text queries, and adjust thresholds to see detections.
161
 
162
  Adapted from LLMDet GitHub repo [Hugging Face demo](https://github.com/iSEE-Laboratory/LLMDet/tree/main/hf_model).
163
 
164
+ This space uses: {model_id}
165
+
166
  See original:
167
  * [LLMDet GitHub](https://github.com/iSEE-Laboratory/LLMDet/tree/main?tab=readme-ov-file)
168
  * [LLMDet Paper](https://arxiv.org/abs/2501.18954) - LLMDet: Learning Strong Open-Vocabulary Object Detectors under the Supervision of Large Language Models
169
+ * LLMDet model checkpoints:
170
+ * [Tiny](https://huggingface.co/fushh7/llmdet_swin_tiny_hf) (173M params)
171
+ * [Base](https://huggingface.co/fushh7/llmdet_swin_base_hf) (233M params)
172
+ * [Large](https://huggingface.co/fushh7/llmdet_swin_large_hf) (344M params)
173
  """
174
  )
175