Update app.py

app.py

@@ -68,7 +68,7 @@ model = Model(confg='./configs/rtdetr/rtdetr_r101vd_6x_coco.yml',ckpt="./checkpo
 #img = Image.open('./a.jpg').convert('RGB').resize((640,640))
 
 
-def detect(img):
+def detect(img,thr):
   #print(img)                    #ndarray
   img = Image.fromarray(img).resize((640,640))
   t_img = transformer(img).unsqueeze(0)#.unsqueeze(0)      #[1,1,640,640]
@@ -77,7 +77,7 @@ def detect(img):
   labels, boxes, scores=model(t_img,size)
   #img_path = Path('./a.jpg')
   draw = ImageDraw.Draw(img)
-  thrh = 0.1
+  thrh = thr
   
   for i in range(t_img.shape[0]):
   
@@ -93,6 +93,6 @@ def detect(img):
   #save_path = Path('./output') / img_path.name
   return img
 
-interface = gr.Interface(fn=detect,inputs="image",outputs="image",title="rt-cope detect")
+interface = gr.Interface(fn=detect,inputs=["image",gr.Slider(label=“thr”, value=0.2, maximum=1, minimum=0)],outputs="image",title="rt-cope detect")
   
 interface.launch()