Upload app.py

app.py

@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+"""app.ipynb
+
+Automatically generated by Colaboratory.
+
+Original file is located at
+    https://colab.research.google.com/drive/1WeNkl1pYnT0qeOTsUFooLFLJ1arRHC00
+"""
+
+import gradio as gr
+from ultralytics import YOLO
+import cv2
+import os
+
+def predict_image(image_input):
+    image = cv2.imread(image_input)
+    # load model
+    model = YOLO("best.pt")
+    #run predict
+    outputs = model.predict(source=image_input)
+    results = output[0].cpu().numpy()
+
+    for i, det in enumerate(results.boxes.xyxy):
+           cv2.rectangle(image, (int(det[0]), int(det[1])), (int(det[2]), int(det[3])), color=(0, 0, 255), thickness=2, lineType=cv2.Line_AA)
+
+    return cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+
+inputs_image = [gr.components.Image(type="filepath", label="Input Image")]
+
+outputs_image = [gr.components.Image(type="numpy", label="Output Image")]
+
+interface_image = gr.Interface(fn = predict_image, inputs=inputs_image, outputs=outputs_image,title="Fire & Smoke Detector", cache_examples=False)
+
+interface_image.launch(Debug=True)