import gradio as gr
from ultralytics import YOLO
# import torch
model = YOLO('best.pt')
def predict(input_image):
"""
Predict model output
"""
price = "0"
return [input_image, price]
with gr.Blocks() as demo:
# Title
gr.Markdown(
"""
AI Cafeteria Price Evaluator
""")
# Model Evaluation
gr.Interface(
fn=predict,
inputs=gr.Image(type="pil"),
outputs=[gr.Image(type="pil", label="Image Prediction"),
gr.Textbox(type="text", label="Price Prediction")]
)
if __name__ == "__main__":
demo.launch()