my frist app

best_int8_openvino_model/best.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:667c192bfb8a38ad7a5fafe62868185de02720203ac793ec36a5cd2d4201c9dc
+size 26059692

best_int8_openvino_model/metadata.yaml

@@ -0,0 +1,15 @@
+description: Ultralytics best model trained on datasets/data.yaml
+author: Ultralytics
+date: '2024-12-24T19:25:07.780379'
+version: 8.3.54
+license: AGPL-3.0 License (https://ultralytics.com/license)
+docs: https://docs.ultralytics.com
+stride: 32
+task: detect
+batch: 1
+imgsz:
+- 640
+- 640
+names:
+  0: mask
+  1: shark

twoobjectdetect.py

@@ -0,0 +1,46 @@
+from ultralytics import YOLO
+from PIL import Image
+import gradio as gr
+from huggingface_hub import snapshot_download
+import os
+
+#model_path = "/Users/markk/Downloads/best_int8_openvino_model"
+#model_path1 = "C:\Users\aungh\Downloads\6319250G\best_int8_openvino_model"
+
+# Define the path to the YOLOv8 mo
+model_path = "./best_int8_openvino_model"   
+
+#Organizations model path location
+MODEL_REPO_ID = "ITI107-2024S2/6319250G"
+
+#load model
+def load_model(repo_id):
+    download_dir = snapshot_download(repo_id)
+    print(download_dir)
+    path  = os.path.join(download_dir, "best_int8_openvino_model")
+    print(path)
+    detection_model = YOLO(path, task='detect')
+    return detection_model
+    
+detection_model = load_model(MODEL_REPO_ID)
+
+#Student ID
+student_info = "Student Id: 6319250G, Name: AUNG HTUN"
+
+#prdeict 
+def predict(pilimg):
+    source = pilimg
+    result = detection_model.predict(source, conf=0.5, iou=0.5)
+    img_bgr = result[0].plot()
+    out_pilimg = Image.fromarray(img_bgr[..., ::-1])  # RGB-order PIL image
+    return out_pilimg
+    
+#UI interface
+gr.Markdown("# Two Object Detection (Shark/Mask)")
+gr.Markdown(student_info)
+gr.Interface(fn=predict,
+             inputs=gr.Image(type="pil",label="Input"),
+             outputs=gr.Image(type="pil",label="Output"),
+             title="Two Object Detection (Shark/Mask)",
+             description=student_info,
+             ).launch(share=True)