Spaces:
Sleeping
Sleeping
from predict import image_identity | |
import gradio as gr | |
import os | |
with gr.Blocks(theme='JohnSmith9982/small_and_pretty') as iface: | |
logo_url = os.path.join(os.path.dirname(__file__), "images/logo.png") | |
# Displaying the logo | |
with gr.Row(elem_classes="center-image"): | |
gr.Image(logo_url, show_label=False, show_download_button=False, | |
show_share_button=False) | |
iface_main = gr.Interface(image_identity, | |
inputs = gr.Image(label="Upload your image"), # Adding a label to the input | |
outputs = [gr.Image(type="numpy", label="Predicted bin"), gr.Textbox(label="Prediction 1"), | |
gr.Textbox(label="Probability"), | |
gr.Textbox(label="Prediction 2"), gr.Textbox(label="Probability"), | |
gr.Textbox(label="Prediction Time")], | |
# Adding a description | |
description="This application is an image recognition app that identifies recyclable objects and provides feedback on the appropriate bin for waste disposal.", | |
examples=[['images/bottle2.jpg'], | |
['images/can2.jpg'], | |
['images/wine2.jpg'], | |
['images/compost2.jpg']], | |
cache_examples=False) # Disabling caching | |
if __name__ == "__main__": | |
# Launch the interface | |
iface.launch(share=False) | |