Update app.py

app.py

@@ -7,9 +7,9 @@ import gradio as gr
 
 cell_detector = YOLO("./weights/yolo_uninfected_cells.pt")
 yolo_detector = YOLO("./weights/yolo_infected_cells.pt")
-# redetr_detector = YOLO("./weights/yolo_infected_cells.pt")
+redetr_detector = YOLO("./weights/redetr_infected_cells.pt")
 
-models = {"Yolo V11": yolo_detector, "Real Time Detection Transformer": yolo_detector}
+models = {"Yolo V11": yolo_detector, "Real Time Detection Transformer": redetr_detector}
 # classes = {"Yolo V11": [0], "Real Time Detection Transformer": [1]}