Update app.py

app.py

@@ -3,8 +3,6 @@ import gradio as gr
 import cv2
 import numpy as np
 from collections import defaultdict
-import os
-import torch
 
 # 初始化模型
 model = None
@@ -12,7 +10,7 @@ model = None
 def load_model():
     global model
     if model is None:
-        model = YOLO('./yolo11x-seg.pt')
+        model = YOLO('./yolo11x-seg.pt')  
     return model
 
 def segment_image(image, conf_threshold, iou_threshold, mask_threshold, line_thickness, use_retina_masks):
@@ -165,4 +163,4 @@ def create_demo():
 
 if __name__ == "__main__":
     demo = create_demo()
-    demo.launch(share=True) 
+    demo.launch()