first commit

app.py

@@ -0,0 +1,61 @@
+import gradio as gr
+import random
+
+class_names = ['cat', 'dog']
+
+def UpdateDropdown(className):
+    class_names.append(className)
+    return gr.Dropdown.update(choices=class_names)
+
+def showPickedClass(className):
+    return className
+
+def image_classifier(inp):
+    if inp is None:
+        return {'cat': 0.3, 'dog': 0.7}
+
+    num_class = len(class_names)
+
+    # Generate random percentages between 0 and 1
+    percentages = [random.random() for _ in range(num_class)]
+    total = sum(percentages)
+
+    # Normalize the percentages to ensure they sum up to 1
+    normalized_percentages = [p / total for p in percentages]
+
+    labeled_result = {name:score for name, score in zip(class_names, normalized_percentages)}
+
+    return labeled_result, gr.Dropdown.update(choices=class_names)
+
+demo = gr.Blocks()
+
+with demo as app:
+    with gr.Row():
+        with gr.Column():
+            inp_img = gr.Image()
+            with gr.Row():
+                clear_btn = gr.Button(value="Clear")
+                process_btn = gr.Button(value="Process", variant="primary")
+        with gr.Column():
+            out_txt = gr.Label(label="Probabilities", num_top_classes=3)
+
+    text_input = gr.Textbox(label="Input the new class here")
+    b1 = gr.Button("Add new class")
+
+    text_options = gr.Dropdown(class_names, label="Class Label", multiselect=False)
+    b2 = gr.Button("Show me the picked class")
+    picked_class = gr.Textbox()
+
+    b1.click(UpdateDropdown, inputs=text_input, outputs=text_options)
+    b2.click(showPickedClass, inputs=text_options, outputs=picked_class)
+
+    process_btn.click(image_classifier, inputs=inp_img, outputs=[out_txt, text_options])
+    clear_btn.click(lambda:(
+        gr.update(value=None),
+        gr.update(value=None)
+        ),
+        inputs=None,
+        outputs=[inp_img, out_txt])
+
+
+demo.launch(debug=True)

requirements.txt

@@ -0,0 +1,63 @@
+aiofiles==23.2.1
+altair==5.1.2
+annotated-types==0.6.0
+anyio==3.7.1
+attrs==23.1.0
+certifi==2023.7.22
+charset-normalizer==3.3.0
+click==8.1.7
+colorama==0.4.6
+contourpy==1.1.1
+cycler==0.12.1
+exceptiongroup==1.1.3
+fastapi==0.103.2
+ffmpy==0.3.1
+filelock==3.12.4
+fonttools==4.43.1
+fsspec==2023.9.2
+gradio==3.47.1
+gradio_client==0.6.0
+h11==0.14.0
+httpcore==0.18.0
+httpx==0.25.0
+huggingface-hub==0.17.3
+idna==3.4
+importlib-resources==6.1.0
+Jinja2==3.1.2
+jsonschema==4.19.1
+jsonschema-specifications==2023.7.1
+kiwisolver==1.4.5
+MarkupSafe==2.1.3
+matplotlib==3.8.0
+mpmath==1.3.0
+networkx==3.1
+numpy==1.26.0
+orjson==3.9.7
+packaging==23.2
+pandas==2.1.1
+Pillow==10.0.1
+pydantic==2.4.2
+pydantic_core==2.10.1
+pydub==0.25.1
+pyparsing==3.1.1
+python-dateutil==2.8.2
+python-multipart==0.0.6
+pytz==2023.3.post1
+PyYAML==6.0.1
+referencing==0.30.2
+requests==2.31.0
+rpds-py==0.10.4
+semantic-version==2.10.0
+six==1.16.0
+sniffio==1.3.0
+starlette==0.27.0
+sympy==1.12
+toolz==0.12.0
+torch==2.1.0
+torchvision==0.16.0
+tqdm==4.66.1
+typing_extensions==4.8.0
+tzdata==2023.3
+urllib3==2.0.6
+uvicorn==0.23.2
+websockets==11.0.3