Create app.py

app.py

@@ -0,0 +1,23 @@
+import numpy as np
+import matplotlib.pyplot as plt
+import random
+import os
+import cv2
+import shutil
+import tqdm
+import glob
+import torch
+from ultralytics import YOLO
+from PIL import Image
+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()