Update app.py

app.py

@@ -73,7 +73,7 @@ model3 =  Model(confg='./configs/rtdetr/rtdetr_r101vd_6x_cocococo.yml',ckpt="./r
 #img = Image.open('./a.jpg').convert('RGB').resize((640,640))
 
 
-def detect(img,thr=0.2,cope='aitod'):
+def detect(img,thr=0.2,trained_dataset='aitod'):
   #print(img)                    #ndarray
   img = Image.fromarray(img).resize((640,640))
   t_img = transformer(img).unsqueeze(0)#.unsqueeze(0)      #[1,1,640,640]
@@ -124,6 +124,6 @@ def detect(img,thr=0.2,cope='aitod'):
   #save_path = Path('./output') / img_path.name
   return img
 
-interface = gr.Interface(fn=detect,inputs=["image",gr.Slider(label="thr", value=0.2, maximum=1, minimum=0),gr.Radio(['aitod','cope12','COCO'])],outputs="image",title="rt-cope detect")
+interface = gr.Interface(fn=detect,inputs=["image",gr.Slider(label="thr", value=0.2, maximum=1, minimum=0),gr.Radio(['aitod','ten_classes','COCO'])],outputs="image",title="rt-cope detect")
   
 interface.launch()