Update app.py

app.py

@@ -72,11 +72,11 @@ def detect_objects(
     device = "cuda" if torch.cuda.is_available() else "cpu"
     model, image_processor = get_model_and_processor(checkpoint)
     model = model.to(device)
-    classes = ["Airplane", "Drone", "Helicopter", "Satellite", "Quadcopter", "Vehicle"]
+    classes = ["person","aeroplane","bicycle","car","motorbike","bus","train","truck","boat"]
     if classes is not None:
         wrong_classes = [cls for cls in classes if cls not in model.config.label2id]
         if wrong_classes:
-            gr.Warning(f"Classes not found in model config: {wrong_classes}")
+            gr.Warning(f"Classes not found in model config")
         keep_ids = [model.config.label2id[cls] for cls in classes if cls in model.config.label2id]
     else:
         keep_ids = None