SmartWaste / app.py
Ariel
Upload files
db12d78
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)