Update app.py

app.py

@@ -22,7 +22,11 @@ pretrain_model_url = {
     'parsing': 'https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/parsing_parsenet.pth',
     'realesrgan': 'https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/RealESRGAN_x2plus.pth'
 }
-
+load_file_from_url(
+    url='https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/codeformer.pth', 
+    model_dir='CodeFormer/weights/CodeFormer', 
+    progress=True
+)
 # Download weights if not already present
 for key, url in pretrain_model_url.items():
     file_path = f"CodeFormer/weights/{key}/{url.split('/')[-1]}"