import gradio as gr
import numpy as np
import joblib
from ultralytics import YOLO

load_model = YOLO("best.pt")

def predict(image):
  result = load_model.predict(source=image, imgsz = 640, conf = 0.25)
  annotated_img = result[0].plot()
  annotated_img = annotated_img[:, :, ::-1]
  return annotated_img

app = gr.Interface(
    fn = predict,
    inputs = gr.Image(type="numpy", label="Upload an image"),
    outputs = gr.Image(type="numpy", label="Detect Brain Tumor"),
    title = "Brain Tumor Detection",
    description="Upload an image. The model will detect and annotate brain tumor."
)

app.launch(share = True)