Update app.py

app.py

@@ -4,6 +4,7 @@ import gradio as gr
 from huggingface_hub import snapshot_download
 import os
 
+#load model
 def load_model(repo_id):
 
     download_dir = snapshot_download(repo_id)
@@ -13,6 +14,7 @@ def load_model(repo_id):
     detection_model = YOLO(model_path, task='detect') 
     return detection_model
 
+#object detection for image 
 def predict(pil_img, conf_thresh, iou_thresh):
     source = pil_img
     results = detection_model.predict(source, conf=conf_thresh, iou=iou_thresh)
@@ -22,6 +24,7 @@ def predict(pil_img, conf_thresh, iou_thresh):
 REPO_ID = "qiqiyuan/glasses_and_mouth" 
 detection_model = load_model(REPO_ID)
 
+#gradio interface
 def iface():
 
     interface = gr.Interface(
@@ -39,5 +42,6 @@ def iface():
 
     return interface
 
-
-iface.launch(share=True) 
+#launch the gradio app
+app_iface = iface()
+app_iface.launch(share=True)