Spaces:

MoinulwithAI
/

FaceAgePredict

Running

MoinulwithAI commited on 28 days ago

Commit

30463cc

verified ·

1 Parent(s): 609997a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ import gradio as gr
 # -------- CONFIG --------
 data_dir = "D:/Dataset/face_age"
-checkpoint_path = "D:/Dataset/age_prediction_model2.pth"
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 print(f"Using device: {device}")

 # -------- CONFIG --------
 data_dir = "D:/Dataset/face_age"
+checkpoint_path = "age_prediction_model2.pth"
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 print(f"Using device: {device}")