Update app.py

app.py

@@ -361,7 +361,7 @@ if uploaded_file is not None:
     checkpoint = torch.load(file_path, map_location=device)
     model.load_state_dict(checkpoint["state_dict"])
     model.eval()
-    image_1 = infer(model, image, 0.5, 0.5, scaled_anchors)
+    image_1 = infer(model, image, 0.7, 0.8, scaled_anchors)
     
     #task 2
     image = Image.open(uploaded_file)
@@ -370,7 +370,7 @@ if uploaded_file is not None:
     checkpoint = torch.load(file_path, map_location=device)
     model.load_state_dict(checkpoint["state_dict"])
     model.eval()
-    image_2 = infer(model, image, 0.5, 0.5, scaled_anchors)
+    image_2 = infer(model, image, 0.7, 0.8, scaled_anchors)
     
     #ft
     image = Image.open(uploaded_file)
@@ -378,7 +378,7 @@ if uploaded_file is not None:
     checkpoint = torch.load(file_path, map_location=device)
     model.load_state_dict(checkpoint["state_dict"])
     model.eval()
-    image_3 = infer(model, image, 0.5, 0.5, scaled_anchors)
+    image_3 = infer(model, image, 0.7, 0.8, scaled_anchors)
     # Streamlit App
     # Widget tải lên file ảnh