Annaig's picture
Update app.py
ff99dcb
import cv2
import gradio as gr
def my_app(img):
grayscale_image = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
classifier = cv2.CascadeClassifier('classifier.xml')
analysedData = classifier.detectMultiScale(grayscale_image,
minSize=(20, 20))
if len(analysedData) != 0:
for (x, y, width, height) in analysedData:
return (cv2.rectangle(img, (x, y),
(x + height, y + width),
(0, 255, 0), 5), "STOP")
else:
return (cv2.imread("thumbs_up.jpg"), "You may continue.")
gr.interface.Interface(fn=my_app, live=True, inputs="image", outputs=["image", "text"]).launch()