Update app.py

app.py

@@ -71,12 +71,12 @@ gender_model.fc = nn.Sequential(
     nn.Dropout(0.4),
     nn.Linear(in_features=512, out_features=2, bias=False)
 )
-age_model.load_state_dict(torch.load('E:/Drive D/barchart/age models/age_mobile_V_1.1/PT/agetest_mobilevit_V_1.1.pt', map_location=torch.device('cpu')))
-gender_model.load_state_dict(torch.load('E:/Drive D/barchart/gender models/gender_mobilevit_version_1.1/pt/gendertest_mobilevit_V_1.1.pt', map_location=torch.device('cpu')))
+age_model.load_state_dict(torch.load('agetest_mobilevit_V_1.1.pt', map_location=torch.device('cpu')))
+gender_model.load_state_dict(torch.load('gendertest_mobilevit_V_1.1.pt', map_location=torch.device('cpu')))
 
 # Load the violence detection model
-violence_model = load_model('E:/TNIBF/modelnew (1).h5')
-MODEL = "E:/TNIBF/yolov8_people.pt"
+violence_model = load_model('modelnew.h5')
+MODEL = "yolov8_people.pt"
 yolo_model = YOLO(MODEL)
 
 # Function to preprocess image for detection