Update app.py

app.py

@@ -60,7 +60,7 @@ def detect_modify(img0, model, conf=0.4, imgsz=640, conf_thres = 0.25, iou_thres
 
     
     # Get names and colors
-    names = model.module.names if hasattr(model 'module') else model.names
+    names = model.module.names if hasattr(models 'module') else model.names
     colors = [[random.randint(0, 255) for _ in range(3)] for _ in names]
 
     # Run inference