Skip to content

Commit

Permalink
Label component now accepts file paths to .json files (#2083)
Browse files Browse the repository at this point in the history
* added label support for file paths

* formatting
  • Loading branch information
abidlabs authored Aug 25, 2022
1 parent 4a1321c commit adf52d2
Show file tree
Hide file tree
Showing 3 changed files with 10 additions and 1 deletion.
4 changes: 3 additions & 1 deletion gradio/components.py
Original file line number Diff line number Diff line change
Expand Up @@ -2753,7 +2753,7 @@ class Label(Changeable, IOComponent, JSONSerializable):
"""
Displays a classification label, along with confidence scores of top categories, if provided.
Preprocessing: this component does *not* accept input.
Postprocessing: expects a {Dict[str, float]} of classes and confidences, or {str} with just the class or an {int}/{float} for regression outputs.
Postprocessing: expects a {Dict[str, float]} of classes and confidences, or {str} with just the class or an {int}/{float} for regression outputs, or a {str} path to a .json file containing a json dictionary in the structure produced by Label.postprocess().
Demos: main_note, titanic_survival
Guides: Gradio_and_ONNX_on_Hugging_Face, image_classification_in_pytorch, image_classification_in_tensorflow, image_classification_with_vision_transformers, building_a_pictionary_app
Expand Down Expand Up @@ -2808,6 +2808,8 @@ def postprocess(self, y: Dict[str, float] | str | float | None) -> Dict | None:
"""
if y is None or y == {}:
return None
if isinstance(y, str) and y.endswith(".json") and os.path.exists(y):
return self.serialize(y)
if isinstance(y, (str, numbers.Number)):
return {"label": str(y)}
if isinstance(y, dict):
Expand Down
6 changes: 6 additions & 0 deletions test/test_components.py
Original file line number Diff line number Diff line change
Expand Up @@ -1289,6 +1289,7 @@ def test_component_functions(self):
)
label_output = gr.Label(num_top_classes=2)
label = label_output.postprocess(y)

self.assertDictEqual(
label,
{
Expand All @@ -1302,6 +1303,11 @@ def test_component_functions(self):
with self.assertRaises(ValueError):
label_output.postprocess([1, 2, 3])

test_file_dir = pathlib.Path(pathlib.Path(__file__).parent, "test_files")
path = str(pathlib.Path(test_file_dir, "test_label_json.json"))
label = label_output.postprocess(path)
self.assertEqual(label["label"], "web site")

self.assertEqual(
label_output.get_config(),
{
Expand Down
1 change: 1 addition & 0 deletions test/test_files/test_label_json.json
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
{"label": "web site", "confidences": [{"label": "web site", "confidence": 0.6625185608863831}, {"label": "envelope", "confidence": 0.02277352288365364}, {"label": "menu", "confidence": 0.012591145932674408}, {"label": "monitor", "confidence": 0.012062293477356434}, {"label": "washer", "confidence": 0.008587697520852089}]}

0 comments on commit adf52d2

Please sign in to comment.