Update app.py

app.py

@@ -36,6 +36,7 @@ if torch.cuda.is_available():
     detector50.model.to('cuda')
     detector101.model.to('cuda')
 
+model = "detr-resnet-101"
 
 COLORS = ["#ff7f7f", "#ff7fbf", "#ff7fff", "#bf7fff",
             "#7f7fff", "#7fbfff", "#7fffff", "#7fffbf",
@@ -69,7 +70,7 @@ def get_figure(in_pil_img):
     
     ax = plt.gca()
     print(f"图像尺寸：{in_pil_img.size}")
-    in_results = query_data(in_pil_img)
+    in_results = query_data(model, in_pil_img)
 
     for prediction in in_results:
         selected_color = choice(COLORS)