import gradio as gr
def function(image):
  model = YOLO('/content/drive/MyDrive/new Movie/best_yolo_2.pt')
  image = np.array(image)
  results = model.predict(image)
  img=Image.fromarray(result[0].plot())
  names_output=[]
  for i in range(num_result):
    name=result[0].names[int(result[0].boxes.cls[i])]
    names_output.append(name)
  return names_output,img
demo=gr.Interface(fn=function,inputs="image",outputs=["text","image"])
demo.launch()