File size: 1,175 Bytes
fc1a76a 0540fd7 fc1a76a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
from transformers import pipeline
import gradio as gr
# Initialize the pipeline
pipe = pipeline("text-classification", model="himanshutitoria/quick-commerce-perishability-model")
# Prediction function
def predict(text):
result = pipe(text)
return result[0]["label"]
# Define the interface
with gr.Blocks() as iface:
# Add an image at the top
gr.Image(
value="top_image.webp", # Replace with your image URL or file path
label="Perishability Tagging",
show_label=False,
)
input_textbox = gr.Textbox(
label="Enter a Product Name",
placeholder="Enter a name here...",
interactive=True,
)
# Add the main Gradio interface
with gr.Row():
gr.Markdown("# Quick Commerce Perishability Tagging")
outputs = gr.Textbox(label="Predicted result")
gr.Interface(
fn=predict,
inputs=input_textbox,
outputs=outputs,
examples=[
["Milk"],
["Perfume"],
["Refined oil"],
["Tata Salt"],
["Cheese"],
["Butterscotch Icecream"],
["Thepla"],
],
)
iface.launch() |