Update app.py

app.py

@@ -23,7 +23,7 @@ def setup_environment():
     model_path = "weights/CodeFormer/codeformer.pth"
     download_file(model_url, model_path)
 
-    # Download facexlib model (for face detection)
+    # Download facelib model (for face detection)
     facelib_url = "https://github.com/xinntao/facexlib/releases/download/v0.1.0/detection_Resnet50_Final.pth"
     facelib_path = "weights/facelib/detection_Resnet50_Final.pth"
     download_file(facelib_url, facelib_path)
@@ -36,7 +36,7 @@ def setup_environment():
 # Load CodeFormer model
 def load_codeformer():
     setup_environment()
-    model = CodeFormer(dim_embd=512, codebook_size=1024, n_head=8, n_layer=9, connect_list=['32', '64', '128', '256'])
+    model = CodeFormer(dim_embd=512, codebook_size=1024, n_head=8, n_layers=9, connect_list=['32', '64', '128', '256'])  # Changed n_layer to n_layers
     model.load_state_dict(torch.load("weights/CodeFormer/codeformer.pth", map_location='cpu'))
     model.eval()
     model = model.to('cpu')  # Force CPU