fastapi added

app.py

@@ -17,9 +17,12 @@ from basicsr.utils.realesrganer import RealESRGANer
 from utils.dataops import auto_split_upscale
 from typing import List, Optional
 
+# Writable directories (Spaces often allow /tmp). Override via env if needed.
+OUTPUT_DIR = os.environ.get("OUTPUT_DIR", "/tmp/face_upscale/output")
+INPUT_DIR  = os.environ.get("INPUT_DIR",  "/tmp/face_upscale/input")
 # Ensure required directories exist at import time (for API mode)
-os.makedirs('output', exist_ok=True)
-os.makedirs('input', exist_ok=True)
+os.makedirs(OUTPUT_DIR, exist_ok=True)
+os.makedirs(INPUT_DIR, exist_ok=True)
 
 # FastAPI imports
 from fastapi import FastAPI, UploadFile, File, Form, Depends, HTTPException, status
@@ -47,31 +50,31 @@ input_images_limit = 5
 # Define URLs and their corresponding local storage paths
 face_models = {
     "GFPGANv1.4.pth": ["https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.4.pth",
-                       "https://github.com/TencentARC/GFPGAN/",
-                       """GFPGAN: Towards Real-World Blind Face Restoration and Upscalling of the image with a Generative Facial Prior.
+                            "https://github.com/TencentARC/GFPGAN/", 
+"""GFPGAN: Towards Real-World Blind Face Restoration and Upscalling of the image with a Generative Facial Prior.
 GFPGAN aims at developing a Practical Algorithm for Real-world Face Restoration.
 It leverages rich and diverse priors encapsulated in a pretrained face GAN (e.g., StyleGAN2) for blind face restoration."""],
     "RestoreFormer++.ckpt": ["https://github.com/wzhouxiff/RestoreFormerPlusPlus/releases/download/v1.0.0/RestoreFormer++.ckpt",
-                            "https://github.com/wzhouxiff/RestoreFormerPlusPlus",
-                            """RestoreFormer++: Towards Real-World Blind Face Restoration from Undegraded Key-Value Pairs.
+                            "https://github.com/wzhouxiff/RestoreFormerPlusPlus", 
+"""RestoreFormer++: Towards Real-World Blind Face Restoration from Undegraded Key-Value Pairs.
 RestoreFormer++ is an extension of RestoreFormer. It proposes to restore a degraded face image with both fidelity and realness by using the powerful fully-spacial attention mechanisms to model the abundant contextual information in the face and its interplay with reconstruction-oriented high-quality priors."""],
     "CodeFormer.pth": ["https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/codeformer.pth",
-                       "https://github.com/sczhou/CodeFormer",
-                       """CodeFormer: Towards Robust Blind Face Restoration with Codebook Lookup Transformer (NeurIPS 2022).
+                            "https://github.com/sczhou/CodeFormer", 
+"""CodeFormer: Towards Robust Blind Face Restoration with Codebook Lookup Transformer (NeurIPS 2022).
 CodeFormer is a Transformer-based model designed to tackle the challenging problem of blind face restoration, where inputs are often severely degraded.
 By framing face restoration as a code prediction task, this approach ensures both improved mapping from degraded inputs to outputs and the generation of visually rich, high-quality faces."""],
     "GPEN-BFR-512.pth": ["https://huggingface.co/akhaliq/GPEN-BFR-512/resolve/main/GPEN-BFR-512.pth",
-                         "https://github.com/yangxy/GPEN",
-                         """GPEN: GAN Prior Embedded Network for Blind Face Restoration in the Wild.
+                            "https://github.com/yangxy/GPEN", 
+"""GPEN: GAN Prior Embedded Network for Blind Face Restoration in the Wild.
 GPEN addresses blind face restoration (BFR) by embedding a GAN into a U-shaped DNN, combining GAN’s ability to generate high-quality images with DNN’s feature extraction.
 This design reconstructs global structure, fine details, and backgrounds from degraded inputs.
 Simple yet effective, GPEN outperforms state-of-the-art methods, delivering realistic results even for severely degraded images."""],
     "GPEN-BFR-1024.pt": ["https://www.modelscope.cn/models/iic/cv_gpen_image-portrait-enhancement-hires/resolve/master/pytorch_model.pt",
-                         "https://www.modelscope.cn/models/iic/cv_gpen_image-portrait-enhancement-hires/files",
-                         """The same as GPEN but for 1024 resolution."""],
+                            "https://www.modelscope.cn/models/iic/cv_gpen_image-portrait-enhancement-hires/files", 
+"""The same as GPEN but for 1024 resolution."""],
     "GPEN-BFR-2048.pt": ["https://www.modelscope.cn/models/iic/cv_gpen_image-portrait-enhancement-hires/resolve/master/pytorch_model-2048.pt",
-                         "https://www.modelscope.cn/models/iic/cv_gpen_image-portrait-enhancement-hires/files",
-                         """The same as GPEN but for 2048 resolution."""],
+                            "https://www.modelscope.cn/models/iic/cv_gpen_image-portrait-enhancement-hires/files", 
+"""The same as GPEN but for 2048 resolution."""],
     "GFPGANv1.3.pth": ["https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth",
                        "https://github.com/TencentARC/GFPGAN/",
                        "The same as GFPGAN but legacy model"],
@@ -85,245 +88,245 @@ Simple yet effective, GPEN outperforms state-of-the-art methods, delivering real
 
 upscale_models = {
     "realesr-general-x4v3.pth": ["https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-x4v3.pth",
-                                 "https://github.com/xinntao/Real-ESRGAN/releases/tag/v0.3.0",
-                                 """Compression Removal, General Upscaler, JPEG, Realistic, Research, Restoration
+                                "https://github.com/xinntao/Real-ESRGAN/releases/tag/v0.3.0", 
+"""Compression Removal, General Upscaler, JPEG, Realistic, Research, Restoration
 xinntao: add realesr-general-x4v3 and realesr-general-wdn-x4v3. They are very tiny models for general scenes, and they may more robust. But as they are tiny models, their performance may be limited."""],
     "realesr-animevideov3.pth": ["https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-animevideov3.pth",
-                                 "https://github.com/xinntao/Real-ESRGAN/releases/tag/v0.2.5.0",
-                                 """Anime, Cartoon, Compression Removal, General Upscaler, JPEG, Realistic, Research, Restoration
+                                "https://github.com/xinntao/Real-ESRGAN/releases/tag/v0.2.5.0", 
+"""Anime, Cartoon, Compression Removal, General Upscaler, JPEG, Realistic, Research, Restoration
 xinntao: update the RealESRGAN AnimeVideo-v3 model, which can achieve better results with a faster inference speed."""],
     "4xLSDIRCompact.pth": ["https://github.com/Phhofm/models/releases/download/4xLSDIRCompact/4xLSDIRCompact.pth",
-                           "https://github.com/Phhofm/models/releases/tag/4xLSDIRCompact",
-                           """Realistic
+                                "https://github.com/Phhofm/models/releases/tag/4xLSDIRCompact", 
+"""Realistic
 Phhofm: Upscale small good quality photos to 4x their size. This is my first ever released self-trained sisr upscaling model."""],
     "4xLSDIRCompactC.pth": ["https://github.com/Phhofm/models/releases/download/4xLSDIRCompactC/4xLSDIRCompactC.pth",
-                            "https://github.com/Phhofm/models/releases/tag/4xLSDIRCompactC",
-                            """Compression Removal, JPEG, Realistic, Restoration
+                                "https://github.com/Phhofm/models/releases/tag/4xLSDIRCompactC", 
+"""Compression Removal, JPEG, Realistic, Restoration
 Phhofm: 4x photo upscaler that handler jpg compression. Trying to extend my previous model to be able to handle compression (JPG 100-30) by manually altering the training dataset, since 4xLSDIRCompact cant handle compression. Use this instead of 4xLSDIRCompact if your photo has compression (like an image from the web)."""],
     "4xLSDIRCompactR.pth": ["https://github.com/Phhofm/models/releases/download/4xLSDIRCompactC/4xLSDIRCompactR.pth",
-                            "https://github.com/Phhofm/models/releases/tag/4xLSDIRCompactC",
-                            """Compression Removal, Realistic, Restoration
+                                "https://github.com/Phhofm/models/releases/tag/4xLSDIRCompactC", 
+"""Compression Removal, Realistic, Restoration
 Phhofm: 4x photo uspcaler that handles jpg compression, noise and slight. Extending my last 4xLSDIRCompact model to Real-ESRGAN, meaning trained on synthetic data instead to handle more kinds of degradations, it should be able to handle compression, noise, and slight blur."""],
     "4xLSDIRCompactN.pth": ["https://github.com/Phhofm/models/releases/download/4xLSDIRCompact3/4xLSDIRCompactC3.pth",
-                            "https://github.com/Phhofm/models/releases/tag/4xLSDIRCompact3",
-                            """Realistic
+                                "https://github.com/Phhofm/models/releases/tag/4xLSDIRCompact3", 
+"""Realistic
 Phhofm: Upscale good quality input photos to x4 their size. The original 4xLSDIRCompact a bit more trained, cannot handle degradation."""],
     "4xLSDIRCompactC3.pth": ["https://github.com/Phhofm/models/releases/download/4xLSDIRCompact3/4xLSDIRCompactC3.pth",
-                             "https://github.com/Phhofm/models/releases/tag/4xLSDIRCompact3",
+                                "https://github.com/Phhofm/models/releases/tag/4xLSDIRCompact3", 
                              """Compression Removal, JPEG, Realistic, Restoration
 Phhofm: Upscale compressed photos to x4 their size. Able to handle JPG compression (30-100)."""],
     "4xLSDIRCompactR3.pth": ["https://github.com/Phhofm/models/releases/download/4xLSDIRCompact3/4xLSDIRCompactR3.pth",
-                             "https://github.com/Phhofm/models/releases/tag/4xLSDIRCompact3",
-                             """Realistic, Restoration
+                                "https://github.com/Phhofm/models/releases/tag/4xLSDIRCompact3", 
+"""Realistic, Restoration
 Phhofm: Upscale (degraded) photos to x4 their size. Trained on synthetic data, meant to handle more degradations."""],
     "4xLSDIRCompactCR3.pth": ["https://github.com/Phhofm/models/releases/download/4xLSDIRCompact3/4xLSDIRCompactCR3.pth",
-                              "https://github.com/Phhofm/models/releases/tag/4xLSDIRCompact3",
-                              """Phhofm: I am releasing the Series 3 from my 4xLSDIRCompact models. In general my suggestion is, if you have good quality input images use 4xLSDIRCompactN3, otherwise try 4xLSDIRCompactC3 which will be able to handle jpg compression and a bit of blur, or then 4xLSDIRCompactCR3, which is an interpolation between C3 and R3 to be able to handle a bit of noise additionally."""],
+                                "https://github.com/Phhofm/models/releases/tag/4xLSDIRCompact3", 
+"""Phhofm: I am releasing the Series 3 from my 4xLSDIRCompact models. In general my suggestion is, if you have good quality input images use 4xLSDIRCompactN3, otherwise try 4xLSDIRCompactC3 which will be able to handle jpg compression and a bit of blur, or then 4xLSDIRCompactCR3, which is an interpolation between C3 and R3 to be able to handle a bit of noise additionally."""],
     "2xParimgCompact.pth": ["https://github.com/Phhofm/models/releases/download/2xParimgCompact/2xParimgCompact.pth",
-                            "https://github.com/Phhofm/models/releases/tag/2xParimgCompact",
-                            """Realistic
+                                "https://github.com/Phhofm/models/releases/tag/2xParimgCompact", 
+"""Realistic
 Phhofm: A 2x photo upscaling compact model based on Microsoft's ImagePairs. This was one of the earliest models I started training and finished it now for release. As can be seen in the examples, this model will affect colors."""],
     "1xExposureCorrection_compact.pth": ["https://github.com/Phhofm/models/releases/download/1xExposureCorrection_compact/1xExposureCorrection_compact.pth",
-                                        "https://github.com/Phhofm/models/releases/tag/1xExposureCorrection_compact",
-                                        """Restoration
+                                         "https://github.com/Phhofm/models/releases/tag/1xExposureCorrection_compact", 
+"""Restoration
 Phhofm: This model is meant as an experiment to see if compact can be used to train on photos to exposure correct those using the pixel, perceptual, color, color and ldl losses. There is no brightness loss. Still it seems to kinda work."""],
     "1xUnderExposureCorrection_compact.pth": ["https://github.com/Phhofm/models/releases/download/1xExposureCorrection_compact/1xUnderExposureCorrection_compact.pth",
-                                             "https://github.com/Phhofm/models/releases/tag/1xExposureCorrection_compact",
-                                             """Restoration
+                                              "https://github.com/Phhofm/models/releases/tag/1xExposureCorrection_compact", 
+"""Restoration
 Phhofm: This model is meant as an experiment to see if compact can be used to train on underexposed images to exposure correct those using the pixel, perceptual, color, color and ldl losses. There is no brightness loss. Still it seems to kinda work."""],
     "1xOverExposureCorrection_compact.pth": ["https://github.com/Phhofm/models/releases/download/1xExposureCorrection_compact/1xOverExposureCorrection_compact.pth",
-                                            "https://github.com/Phhofm/models/releases/tag/1xExposureCorrection_compact",
-                                            """Restoration
+                                             "https://github.com/Phhofm/models/releases/tag/1xExposureCorrection_compact", 
+"""Restoration
 Phhofm: This model is meant as an experiment to see if compact can be used to train on overexposed images to exposure correct those using the pixel, perceptual, color, color and ldl losses. There is no brightness loss. Still it seems to kinda work."""],
     "2x-sudo-UltraCompact.pth": ["https://objectstorage.us-phoenix-1.oraclecloud.com/n/ax6ygfvpvzka/b/open-modeldb-files/o/2x-sudo-UltraCompact.pth",
-                                 "https://openmodeldb.info/models/2x-sudo-UltraCompact",
-                                 """Anime, Cartoon, Restoration
+                                "https://openmodeldb.info/models/2x-sudo-UltraCompact", 
+"""Anime, Cartoon, Restoration
 sudo: Realtime animation restauration and doing stuff like deblur and compression artefact removal."""],
     "2x_AnimeJaNai_HD_V3_SuperUltraCompact.pth": ["https://github.com/the-database/mpv-upscale-2x_animejanai/releases/download/3.0.0/2x_AnimeJaNai_HD_V3_ModelsOnly.zip",
-                                                  "https://openmodeldb.info/models/2x-AnimeJaNai-HD-V3-SuperUltraCompact",
-                                                  """Anime, Compression Removal, Restoration
+                                                  "https://openmodeldb.info/models/2x-AnimeJaNai-HD-V3-SuperUltraCompact", 
+"""Anime, Compression Removal, Restoration
 the-database: Real-time 2x Real-ESRGAN Compact/UltraCompact/SuperUltraCompact models designed for upscaling 1080p anime to 4K."""],
     "2x_AnimeJaNai_HD_V3_UltraCompact.pth": ["https://github.com/the-database/mpv-upscale-2x_animejanai/releases/download/3.0.0/2x_AnimeJaNai_HD_V3_ModelsOnly.zip",
-                                             "https://openmodeldb.info/models/2x-AnimeJaNai-HD-V3-UltraCompact",
-                                             """Anime, Compression Removal, Restoration
+                                             "https://openmodeldb.info/models/2x-AnimeJaNai-HD-V3-UltraCompact", 
+"""Anime, Compression Removal, Restoration
 the-database: Real-time 2x Real-ESRGAN Compact/UltraCompact/SuperUltraCompact models designed for upscaling 1080p anime to 4K."""],
     "2x_AnimeJaNai_HD_V3_Compact.pth": ["https://github.com/the-database/mpv-upscale-2x_animejanai/releases/download/3.0.0/2x_AnimeJaNai_HD_V3_ModelsOnly.zip",
-                                        "https://openmodeldb.info/models/2x-AnimeJaNai-HD-V3-Compact",
-                                        """Anime, Compression Removal, Restoration
+                                                  "https://openmodeldb.info/models/2x-AnimeJaNai-HD-V3-Compact", 
+"""Anime, Compression Removal, Restoration
 the-database: Real-time 2x Real-ESRGAN Compact/UltraCompact/SuperUltraCompact models designed for upscaling 1080p anime to 4K."""],
     "RealESRGAN_x4plus_anime_6B.pth": ["https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth",
-                                       "https://github.com/xinntao/Real-ESRGAN/releases/tag/v0.2.2.4",
-                                       """Anime, Cartoon, Compression Removal, General Upscaler, JPEG, Realistic, Research, Restoration
+                                      "https://github.com/xinntao/Real-ESRGAN/releases/tag/v0.2.2.4", 
+"""Anime, Cartoon, Compression Removal, General Upscaler, JPEG, Realistic, Research, Restoration
 xinntao: We add RealESRGAN_x4plus_anime_6B.pth, which is optimized for anime images with much smaller model size."""],
     "RealESRGAN_x2plus.pth": ["https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.1/RealESRGAN_x2plus.pth",
-                              "https://github.com/xinntao/Real-ESRGAN/releases/tag/v0.2.1",
-                              """Compression Removal, General Upscaler, JPEG, Realistic, Research, Restoration
+                                      "https://github.com/xinntao/Real-ESRGAN/releases/tag/v0.2.1", 
+"""Compression Removal, General Upscaler, JPEG, Realistic, Research, Restoration
 xinntao: Add RealESRGAN_x2plus.pth model"""],
     "RealESRNet_x4plus.pth": ["https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.1/RealESRNet_x4plus.pth",
-                              "https://github.com/xinntao/Real-ESRGAN/releases/tag/v0.1.1",
-                              """Compression Removal, General Upscaler, JPEG, Realistic, Research, Restoration
+                                      "https://github.com/xinntao/Real-ESRGAN/releases/tag/v0.1.1", 
+"""Compression Removal, General Upscaler, JPEG, Realistic, Research, Restoration
 xinntao: This release is mainly for storing pre-trained models and executable files."""],
     "RealESRGAN_x4plus.pth": ["https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth",
-                              "https://github.com/xinntao/Real-ESRGAN/releases/tag/v0.1.0",
-                              """Compression Removal, General Upscaler, JPEG, Realistic, Research, Restoration
+                                      "https://github.com/xinntao/Real-ESRGAN/releases/tag/v0.1.0", 
+"""Compression Removal, General Upscaler, JPEG, Realistic, Research, Restoration
 xinntao: This release is mainly for storing pre-trained models and executable files."""],
     "4x-AnimeSharp.pth": ["https://huggingface.co/utnah/esrgan/resolve/main/4x-AnimeSharp.pth?download=true",
-                          "https://openmodeldb.info/models/4x-AnimeSharp",
-                          """Anime, Cartoon, Text
+                         "https://openmodeldb.info/models/4x-AnimeSharp", 
+"""Anime, Cartoon, Text
 Kim2091: Interpolation between 4x-UltraSharp and 4x-TextSharp-v0.5. Works amazingly on anime."""],
     "4x_IllustrationJaNai_V1_ESRGAN_135k.pth": ["https://drive.google.com/uc?export=download&confirm=1&id=1qpioSqBkB_IkSBhEAewSSNFt6qgkBimP",
-                                               "https://openmodeldb.info/models/4x-IllustrationJaNai-V1-DAT2",
-                                               """Anime, Cartoon, Compression Removal, Dehalftone, General Upscaler, JPEG, Manga, Restoration
+                                               "https://openmodeldb.info/models/4x-IllustrationJaNai-V1-DAT2", 
+"""Anime, Cartoon, Compression Removal, Dehalftone, General Upscaler, JPEG, Manga, Restoration
 the-database: Model for color images including manga covers and color illustrations, digital art, visual novel art, artbooks, and more."""],
     "2x-sudo-RealESRGAN.pth": ["https://objectstorage.us-phoenix-1.oraclecloud.com/n/ax6ygfvpvzka/b/open-modeldb-files/o/2x-sudo-RealESRGAN.pth",
-                               "https://openmodeldb.info/models/2x-sudo-RealESRGAN",
-                               """Anime, Cartoon
+                               "https://openmodeldb.info/models/2x-sudo-RealESRGAN", 
+"""Anime, Cartoon
 sudo: Tried to make the best 2x model there is for drawings."""],
     "2x-sudo-RealESRGAN-Dropout.pth": ["https://objectstorage.us-phoenix-1.oraclecloud.com/n/ax6ygfvpvzka/b/open-modeldb-files/o/2x-sudo-RealESRGAN-Dropout.pth",
-                                       "https://openmodeldb.info/models/2x-sudo-RealESRGAN-Dropout",
-                                       """Anime, Cartoon
+                               "https://openmodeldb.info/models/2x-sudo-RealESRGAN-Dropout", 
+"""Anime, Cartoon
 sudo: Tried to make the best 2x model there is for drawings."""],
     "4xNomos2_otf_esrgan.pth": ["https://github.com/Phhofm/models/releases/download/4xNomos2_otf_esrgan/4xNomos2_otf_esrgan.pth",
-                                "https://github.com/Phhofm/models/releases/tag/4xNomos2_otf_esrgan",
-                                """Compression Removal, JPEG, Realistic, Restoration
+                               "https://github.com/Phhofm/models/releases/tag/4xNomos2_otf_esrgan", 
+"""Compression Removal, JPEG, Realistic, Restoration
 Phhofm: Restoration, 4x ESRGAN model for photography, trained using the Real-ESRGAN otf degradation pipeline."""],
     "4xNomosWebPhoto_esrgan.pth": ["https://github.com/Phhofm/models/releases/download/4xNomosWebPhoto_esrgan/4xNomosWebPhoto_esrgan.pth",
-                                   "https://github.com/Phhofm/models/releases/tag/4xNomosWebPhoto_esrgan",
-                                   """Realistic, Restoration
+                               "https://github.com/Phhofm/models/releases/tag/4xNomosWebPhoto_esrgan", 
+"""Realistic, Restoration
 Phhofm: Restoration, 4x ESRGAN model for photography, trained with realistic noise, lens blur, jpg and webp re-compression."""],
     "4x_foolhardy_Remacri.pth": ["https://civitai.com/api/download/models/164821?type=Model&format=PickleTensor",
-                                 "https://openmodeldb.info/models/4x-Remacri",
-                                 """Original
+                               "https://openmodeldb.info/models/4x-Remacri", 
+"""Original
 FoolhardyVEVO: A creation of BSRGAN with more details and less smoothing."""],
     "4x_foolhardy_Remacri_ExtraSmoother.pth": ["https://civitai.com/api/download/models/164822?type=Model&format=PickleTensor",
-                                              "https://openmodeldb.info/models/4x-Remacri",
-                                              """ExtraSmoother
+                               "https://openmodeldb.info/models/4x-Remacri", 
+"""ExtraSmoother
 FoolhardyVEVO: A creation of BSRGAN with more details and less smoothing."""],
     "4xNomos8kDAT.pth": ["https://github.com/Phhofm/models/releases/download/4xNomos8kDAT/4xNomos8kDAT.pth",
-                         "https://openmodeldb.info/models/4x-Nomos8kDAT",
-                         """Anime, Compression Removal, General Upscaler, JPEG, Realistic, Restoration
+                                             "https://openmodeldb.info/models/4x-Nomos8kDAT", 
+"""Anime, Compression Removal, General Upscaler, JPEG, Realistic, Restoration
 Phhofm: A 4x photo upscaler with otf jpg compression, blur and resize, trained on musl's Nomos8k_sfw dataset for realisic sr."""],
     "4x-DWTP-DS-dat2-v3.pth": ["https://objectstorage.us-phoenix-1.oraclecloud.com/n/ax6ygfvpvzka/b/open-modeldb-files/o/4x-DWTP-DS-dat2-v3.pth",
-                               "https://openmodeldb.info/models/4x-DWTP-DS-dat2-v3",
-                               """Dehalftone, Restoration
+                                             "https://openmodeldb.info/models/4x-DWTP-DS-dat2-v3", 
+"""Dehalftone, Restoration
 umzi.x.dead: DAT descreenton model, designed to reduce discrepancies on tiles due to too much loss of the first version."""],
     "4xBHI_dat2_real.pth": ["https://github.com/Phhofm/models/releases/download/4xBHI_dat2_real/4xBHI_dat2_real.pth",
-                            "https://github.com/Phhofm/models/releases/tag/4xBHI_dat2_real",
-                            """Compression Removal, JPEG, Realistic
+                                             "https://github.com/Phhofm/models/releases/tag/4xBHI_dat2_real", 
+"""Compression Removal, JPEG, Realistic
 Phhofm: 4x dat2 upscaling model for web and realistic images."""],
     "4xBHI_dat2_otf.pth": ["https://github.com/Phhofm/models/releases/download/4xBHI_dat2_otf/4xBHI_dat2_otf.pth",
-                           "https://github.com/Phhofm/models/releases/tag/4xBHI_dat2_otf",
-                           """Compression Removal, JPEG
+                                             "https://github.com/Phhofm/models/releases/tag/4xBHI_dat2_otf", 
+"""Compression Removal, JPEG
 Phhofm: 4x dat2 upscaling model, trained with the real-esrgan otf pipeline on my bhi dataset."""],
     "4xBHI_dat2_multiblur.pth": ["https://github.com/Phhofm/models/releases/download/4xBHI_dat2_multiblurjpg/4xBHI_dat2_multiblur.pth",
-                                "https://github.com/Phhofm/models/releases/tag/4xBHI_dat2_multiblurjpg",
-                                """Phhofm: the 4xBHI_dat2_multiblur checkpoint (trained to 250000 iters), which cannot handle compression but might give just slightly better output on non-degraded input."""],
+                                             "https://github.com/Phhofm/models/releases/tag/4xBHI_dat2_multiblurjpg", 
+"""Phhofm: the 4xBHI_dat2_multiblur checkpoint (trained to 250000 iters), which cannot handle compression but might give just slightly better output on non-degraded input."""],
     "4xBHI_dat2_multiblurjpg.pth": ["https://github.com/Phhofm/models/releases/download/4xBHI_dat2_multiblurjpg/4xBHI_dat2_multiblurjpg.pth",
-                                   "https://github.com/Phhofm/models/releases/tag/4xBHI_dat2_multiblurjpg",
-                                   """Compression Removal, JPEG
+                                             "https://github.com/Phhofm/models/releases/tag/4xBHI_dat2_multiblurjpg", 
+"""Compression Removal, JPEG
 Phhofm: 4x dat2 upscaling model, trained with down_up,linear, cubic_mitchell, lanczos, gauss and box scaling algos, some average, gaussian and anisotropic blurs and jpg compression."""],
     "4x_IllustrationJaNai_V1_DAT2_190k.pth": ["https://drive.google.com/uc?export=download&confirm=1&id=1qpioSqBkB_IkSBhEAewSSNFt6qgkBimP",
-                                             "https://openmodeldb.info/models/4x-IllustrationJaNai-V1-DAT2",
-                                             """Anime, Cartoon, Compression Removal, Dehalftone, General Upscaler, JPEG, Manga, Restoration
+                                             "https://openmodeldb.info/models/4x-IllustrationJaNai-V1-DAT2", 
+"""Anime, Cartoon, Compression Removal, Dehalftone, General Upscaler, JPEG, Manga, Restoration
 the-database: Model for color images including manga covers and color illustrations, digital art, visual novel art, artbooks, and more."""],
     "4x-PBRify_UpscalerDAT2_V1.pth": ["https://github.com/Kim2091/Kim2091-Models/releases/download/4x-PBRify_UpscalerDAT2_V1/4x-PBRify_UpscalerDAT2_V1.pth",
-                                      "https://github.com/Kim2091/Kim2091-Models/releases/tag/4x-PBRify_UpscalerDAT2_V1",
-                                      """Compression Removal, DDS, Game Textures, Restoration
+                                      "https://github.com/Kim2091/Kim2091-Models/releases/tag/4x-PBRify_UpscalerDAT2_V1", 
+"""Compression Removal, DDS, Game Textures, Restoration
 Kim2091: Yet another model in the PBRify_Remix series."""],
     "4xBHI_dat2_otf_nn.pth": ["https://github.com/Phhofm/models/releases/download/4xBHI_dat2_otf_nn/4xBHI_dat2_otf_nn.pth",
-                              "https://github.com/Phhofm/models/releases/tag/4xBHI_dat2_otf_nn",
-                              """Compression Removal, JPEG
+                              "https://github.com/Phhofm/models/releases/tag/4xBHI_dat2_otf_nn", 
+"""Compression Removal, JPEG
 Phhofm: 4x dat2 upscaling model, trained with the real-esrgan otf pipeline but without noise, on my bhi dataset."""],
     "4xNomos8kSCHAT-L.pth": ["https://github.com/Phhofm/models/releases/download/4xNomos8kSCHAT/4xNomos8kSCHAT-L.pth",
-                             "https://openmodeldb.info/models/4x-Nomos8kSCHAT-L",
-                             """Anime, Compression Removal, General Upscaler, JPEG, Realistic, Restoration
+                              "https://openmodeldb.info/models/4x-Nomos8kSCHAT-L", 
+"""Anime, Compression Removal, General Upscaler, JPEG, Realistic, Restoration
 Phhofm: 4x photo upscaler with otf jpg compression and blur, trained on musl's Nomos8k_sfw dataset for realisic sr."""],
     "4xNomos8kSCHAT-S.pth": ["https://github.com/Phhofm/models/releases/download/4xNomos8kSCHAT/4xNomos8kSCHAT-S.pth",
-                             "https://openmodeldb.info/models/4x-Nomos8kSCHAT-S",
-                             """Anime, Compression Removal, General Upscaler, JPEG, Realistic, Restoration
+                              "https://openmodeldb.info/models/4x-Nomos8kSCHAT-S", 
+"""Anime, Compression Removal, General Upscaler, JPEG, Realistic, Restoration
 Phhofm: 4x photo upscaler with otf jpg compression and blur, trained on musl's Nomos8k_sfw dataset for realisic sr."""],
     "4xNomos8kHAT-L_otf.pth": ["https://github.com/Phhofm/models/releases/download/4xNomos8kHAT-L_otf/4xNomos8kHAT-L_otf.pth",
-                               "https://openmodeldb.info/models/4x-Nomos8kHAT-L-otf",
-                               """Faces, General Upscaler, Realistic, Restoration
+                              "https://openmodeldb.info/models/4x-Nomos8kHAT-L-otf", 
+"""Faces, General Upscaler, Realistic, Restoration
 Phhofm: 4x photo upscaler trained with otf, handles some jpg compression, some blur and some noise."""],
     "4xBHI_small_hat-l.pth": ["https://github.com/Phhofm/models/releases/download/4xBHI_small_hat-l/4xBHI_small_hat-l.pth",
-                              "https://github.com/Phhofm/models/releases/tag/4xBHI_small_hat-l",
+                              "https://github.com/Phhofm/models/releases/tag/4xBHI_small_hat-l", 
                               """Phhofm: 4x hat-l upscaling model for good quality input. This model does not handle any degradations."""],
     "4xHFA2k_ludvae_realplksr_dysample.pth": ["https://github.com/Phhofm/models/releases/download/4xHFA2k_ludvae_realplksr_dysample/4xHFA2k_ludvae_realplksr_dysample.pth",
-                                             "https://openmodeldb.info/models/4x-HFA2k-ludvae-realplksr-dysample",
-                                             """Anime, Compression Removal, Restoration
+                                             "https://openmodeldb.info/models/4x-HFA2k-ludvae-realplksr-dysample", 
+"""Anime, Compression Removal, Restoration
 Phhofm: A Dysample RealPLKSR 4x upscaling model for anime single-image resolution."""],
     "4xArtFaces_realplksr_dysample.pth": ["https://github.com/Phhofm/models/releases/download/4xArtFaces_realplksr_dysample/4xArtFaces_realplksr_dysample.pth",
-                                         "https://openmodeldb.info/models/4x-ArtFaces-realplksr-dysample",
-                                         """ArtFaces
+                                             "https://openmodeldb.info/models/4x-ArtFaces-realplksr-dysample", 
+"""ArtFaces
 Phhofm: A Dysample RealPLKSR 4x upscaling model for art / painted faces."""],
     "4x-PBRify_RPLKSRd_V3.pth": ["https://github.com/Kim2091/Kim2091-Models/releases/download/4x-PBRify_RPLKSRd_V3/4x-PBRify_RPLKSRd_V3.pth",
-                                 "https://github.com/Kim2091/Kim2091-Models/releases/tag/4x-PBRify_RPLKSRd_V3",
-                                 """Compression Removal, DDS, Debanding, Dedither, Dehalo, Game Textures, Restoration
+                                             "https://github.com/Kim2091/Kim2091-Models/releases/tag/4x-PBRify_RPLKSRd_V3", 
+"""Compression Removal, DDS, Debanding, Dedither, Dehalo, Game Textures, Restoration
 Kim2091: This update brings a new upscaling model, 4x-PBRify_RPLKSRd_V3."""],
     "4xNomos2_realplksr_dysample.pth": ["https://github.com/Phhofm/models/releases/download/4xNomos2_realplksr_dysample/4xNomos2_realplksr_dysample.pth",
-                                        "https://openmodeldb.info/models/4x-Nomos2-realplksr-dysample",
-                                        """Compression Removal, JPEG, Realistic, Restoration
+                                             "https://openmodeldb.info/models/4x-Nomos2-realplksr-dysample", 
+"""Compression Removal, JPEG, Realistic, Restoration
 Phhofm: A Dysample RealPLKSR 4x upscaling model that was trained with / handles jpg compression down to 70 on the Nomosv2 dataset."""],
     "2x-AnimeSharpV2_RPLKSR_Sharp.pth": ["https://github.com/Kim2091/Kim2091-Models/releases/download/2x-AnimeSharpV2_Set/2x-AnimeSharpV2_RPLKSR_Sharp.pth",
-                                        "https://github.com/Kim2091/Kim2091-Models/releases/tag/2x-AnimeSharpV2_Set",
-                                        """Anime, Compression Removal, Restoration
+                                        "https://github.com/Kim2091/Kim2091-Models/releases/tag/2x-AnimeSharpV2_Set", 
+"""Anime, Compression Removal, Restoration
 Kim2091: This is my first anime model in years. Sharp: For heavily degraded sources."""],
     "2x-AnimeSharpV2_RPLKSR_Soft.pth": ["https://github.com/Kim2091/Kim2091-Models/releases/download/2x-AnimeSharpV2_Set/2x-AnimeSharpV2_RPLKSR_Soft.pth",
-                                        "https://github.com/Kim2091/Kim2091-Models/releases/tag/2x-AnimeSharpV2_Set",
-                                        """Anime, Compression Removal, Restoration
+                                         "https://github.com/Kim2091/Kim2091-Models/releases/tag/2x-AnimeSharpV2_Set", 
+"""Anime, Compression Removal, Restoration
 Kim2091: This is my first anime model in years. Soft: For cleaner sources."""],
     "4xPurePhoto-RealPLSKR.pth": ["https://github.com/starinspace/StarinspaceUpscale/releases/download/Models/4xPurePhoto-RealPLSKR.pth",
-                                  "https://openmodeldb.info/models/4x-PurePhoto-RealPLSKR",
-                                  """AI Generated, Compression Removal, JPEG, Realistic, Restoration
+                                        "https://openmodeldb.info/models/4x-PurePhoto-RealPLSKR", 
+"""AI Generated, Compression Removal, JPEG, Realistic, Restoration
 asterixcool: Skilled in working with cats, hair, parties, and creating clear images."""],
     "2x_Text2HD_v.1-RealPLKSR.pth": ["https://github.com/starinspace/StarinspaceUpscale/releases/download/Models/2x_Text2HD_v.1-RealPLKSR.pth",
-                                     "https://openmodeldb.info/models/2x-Text2HD-v-1",
-                                     """Compression Removal, Denoise, General Upscaler, JPEG, Restoration, Text
+                                        "https://openmodeldb.info/models/2x-Text2HD-v-1", 
+"""Compression Removal, Denoise, General Upscaler, JPEG, Restoration, Text
 asterixcool: The upscale model is specifically designed to enhance lower-quality text images."""],
     "2xVHS2HD-RealPLKSR.pth": ["https://github.com/starinspace/StarinspaceUpscale/releases/download/Models/2xVHS2HD-RealPLKSR.pth",
-                                "https://openmodeldb.info/models/2x-VHS2HD",
-                                """Compression Removal, Dehalo, Realistic, Restoration, Video Frame
+                                        "https://openmodeldb.info/models/2x-VHS2HD", 
+"""Compression Removal, Dehalo, Realistic, Restoration, Video Frame
 asterixcool: An advanced VHS recording model designed to enhance video quality by reducing artifacts."""],
     "4xNomosWebPhoto_RealPLKSR.pth": ["https://github.com/Phhofm/models/releases/download/4xNomosWebPhoto_RealPLKSR/4xNomosWebPhoto_RealPLKSR.pth",
-                                     "https://openmodeldb.info/models/4x-NomosWebPhoto-RealPLKSR",
-                                     """Realistic, Restoration
+                                        "https://openmodeldb.info/models/4x-NomosWebPhoto-RealPLKSR", 
+"""Realistic, Restoration
 Phhofm: 4x RealPLKSR model for photography, trained with realistic noise, lens blur, jpg and webp re-compression."""],
     "4xNomos2_hq_drct-l.pth": ["https://github.com/Phhofm/models/releases/download/4xNomos2_hq_drct-l/4xNomos2_hq_drct-l.pth",
-                               "https://github.com/Phhofm/models/releases/tag/4xNomos2_hq_drct-l",
-                               """General Upscaler, Realistic
+                                        "https://github.com/Phhofm/models/releases/tag/4xNomos2_hq_drct-l",
+"""General Upscaler, Realistic
 Phhofm: An drct-l 4x upscaling model, similiar to the 4xNomos2_hq_atd, 4xNomos2_hq_dat2 and 4xNomos2_hq_mosr models."""],
     "4xNomos2_hq_atd.pth": ["https://github.com/Phhofm/models/releases/download/4xNomos2_hq_atd/4xNomos2_hq_atd.pth",
-                            "https://github.com/Phhofm/models/releases/tag/4xNomos2_hq_atd",
-                            """General Upscaler, Realistic
+                                         "https://github.com/Phhofm/models/releases/tag/4xNomos2_hq_atd",
+"""General Upscaler, Realistic
 Phhofm: An atd 4x upscaling model, similiar to the 4xNomos2_hq_dat2 or 4xNomos2_hq_mosr models."""],
     "4xNomos2_hq_mosr.pth": ["https://github.com/Phhofm/models/releases/download/4xNomos2_hq_mosr/4xNomos2_hq_mosr.pth",
-                             "https://github.com/Phhofm/models/releases/tag/4xNomos2_hq_mosr",
-                             """General Upscaler, Realistic
+                                         "https://github.com/Phhofm/models/releases/tag/4xNomos2_hq_mosr",
+"""General Upscaler, Realistic
 Phhofm: A 4x MoSR upscaling model, meant for non-degraded input."""],
     "2x-AnimeSharpV2_MoSR_Sharp.pth": ["https://github.com/Kim2091/Kim2091-Models/releases/download/2x-AnimeSharpV2_Set/2x-AnimeSharpV2_MoSR_Sharp.pth",
-                                      "https://github.com/Kim2091/Kim2091-Models/releases/tag/2x-AnimeSharpV2_Set",
-                                      """Anime, Compression Removal, Restoration
+                                         "https://github.com/Kim2091/Kim2091-Models/releases/tag/2x-AnimeSharpV2_Set",
+"""Anime, Compression Removal, Restoration
 Kim2091: This is my first anime model in years. Sharp: For heavily degraded sources."""],
     "2x-AnimeSharpV2_MoSR_Soft.pth": ["https://github.com/Kim2091/Kim2091-Models/releases/download/2x-AnimeSharpV2_Set/2x-AnimeSharpV2_MoSR_Soft.pth",
-                                     "https://github.com/Kim2091/Kim2091-Models/releases/tag/2x-AnimeSharpV2_Set",
-                                     """Anime, Compression Removal, Restoration
+                                         "https://github.com/Kim2091/Kim2091-Models/releases/tag/2x-AnimeSharpV2_Set",
+"""Anime, Compression Removal, Restoration
 Kim2091: This is my first anime model in years. Soft: For cleaner sources."""],
     "4xNomos8kSCSRFormer.pth": ["https://github.com/Phhofm/models/releases/download/4xNomos8kSCSRFormer/4xNomos8kSCSRFormer.pth",
-                                "https://github.com/Phhofm/models/releases/tag/4xNomos8kSCSRFormer",
-                                """Anime, Compression Removal, General Upscaler, JPEG, Realistic, Restoration
+                                             "https://github.com/Phhofm/models/releases/tag/4xNomos8kSCSRFormer",
+"""Anime, Compression Removal, General Upscaler, JPEG, Realistic, Restoration
 Phhofm: 4x photo upscaler with otf jpg compression and blur, trained on musl's Nomos8k_sfw dataset for realisic sr."""],
     "4xFrankendataFullDegradation_SRFormer460K_g.pth": ["https://drive.google.com/uc?export=download&confirm=1&id=1PZrj-8ofxhORv_OgTVSoRt3dYi-BtiDj",
-                                                       "https://openmodeldb.info/models/4x-Frankendata-FullDegradation-SRFormer",
-                                                       """Compression Removal, Denoise, Realistic, Restoration
+                                                    "https://openmodeldb.info/models/4x-Frankendata-FullDegradation-SRFormer",
+"""Compression Removal, Denoise, Realistic, Restoration
 Crustaceous D: 4x realistic upscaler that may also work for general purpose usage."""],
     "4xFrankendataPretrainer_SRFormer400K_g.pth": ["https://drive.google.com/uc?export=download&confirm=1&id=1SaKvpYYIm2Vj2m9GifUMlNCbmkE6JZmr",
-                                                  "https://openmodeldb.info/models/4x-FrankendataPretainer-SRFormer",
-                                                  """Realistic, Restoration
+                                                    "https://openmodeldb.info/models/4x-FrankendataPretainer-SRFormer",
+"""Realistic, Restoration
 Crustaceous D: 4x realistic upscaler that may also work for general purpose usage."""],
     "1xFrankenfixer_SRFormerLight_g.pth": ["https://drive.google.com/uc?export=download&confirm=1&id=1UJ0iyFn4IGNhPIgNgrQrBxYsdDloFc9I",
-                                          "https://openmodeldb.info/models/1x-Frankenfixer-SRFormerLight",
-                                          """Realistic, Restoration
+                                                  "https://openmodeldb.info/models/1x-Frankenfixer-SRFormerLight",
+"""Realistic, Restoration
 Crustaceous D: A 1x model designed to reduce artifacts and restore detail to images upscaled by 4xFrankendata_FullDegradation_SRFormer."""],
 }
 
@@ -415,7 +418,7 @@ class Upscale:
                 if "layers.0.blocks.0.attn.attns.0.rpe_biases" in loadnet:
                     split_sizes = loadnet["layers.0.blocks.0.attn.attns.0.rpe_biases"][-1] + 1
                     split_size = [int(x) for x in split_sizes]
-                model = DAT(img_size=img_size, in_chans=in_chans, embed_dim=embed_dim, split_size=split_size, depth=depth, num_heads=num_heads, expansion_factor=expansion_factor,
+                model = DAT(img_size=img_size, in_chans=in_chans, embed_dim=embed_dim, split_size=split_size, depth=depth, num_heads=num_heads, expansion_factor=expansion_factor, 
                             qkv_bias=qkv_bias, resi_connection=resi_connection, upsampler=upsampler, upscale=netscale)
             elif upscale_type == "HAT":
                 half = False
@@ -473,7 +476,7 @@ class Upscale:
                 ape = "absolute_pos_embed" in loadnet
                 model = DRCT(in_chans=in_chans, img_size=64, window_size=window_size, compress_ratio=3, squeeze_factor=30,
                              conv_scale=0.01, overlap_ratio=0.5, img_range=1., depths=depths, embed_dim=embed_dim, num_heads=num_heads,
-                             mlp_ratio=mlp_ratio, qkv_bias=qkv_bias, ape=ape, patch_norm=patch_norm, use_checkpoint=False,
+                    mlp_ratio=mlp_ratio, qkv_bias=qkv_bias, ape=ape, patch_norm=patch_norm, use_checkpoint=False,
                              upscale=netscale, upsampler=upsampler, resi_connection=resi_connection, gc=gc_adjust1)
             elif upscale_type == "ATD":
                 half = False
@@ -560,7 +563,7 @@ class Upscale:
                         if nice_number == patches_resolution:
                             img_size = nice_number
                             break
-                model = SRFormer(img_size=img_size, in_chans=in_chans, embed_dim=embed_dim, depths=depths, num_heads=num_heads, window_size=window_size, mlp_ratio=mlp_ratio,
+                model = SRFormer(img_size=img_size, in_chans=in_chans, embed_dim=embed_dim, depths=depths, num_heads=num_heads, window_size=window_size, mlp_ratio=mlp_ratio, 
                                  qkv_bias=qkv_bias, qk_scale=None, ape=ape, patch_norm=patch_norm, upscale=netscale, upsampler=upsampler, resi_connection=resi_connection)
         if model:
             self.realesrganer = RealESRGANer(scale=netscale, model_path=os.path.join("weights", "upscale", upscale_model), model=model, tile=0, tile_pad=10, pre_pad=0, half=half)
@@ -662,13 +665,13 @@ class Upscale:
                 self.modelInUse = ""
             files = []
             unique_id = str(int(time.time()))
-            zip_cropf_path = f"output/{unique_id}_cropped_faces{self.modelInUse}.zip"
+            zip_cropf_path = os.path.join(OUTPUT_DIR, f"{unique_id}_cropped_faces{self.modelInUse}.zip")
             zipf_cropf = zipfile.ZipFile(zip_cropf_path, 'w', zipfile.ZIP_DEFLATED)
-            zip_restoref_path = f"output/{unique_id}_restored_faces{self.modelInUse}.zip"
+            zip_restoref_path = os.path.join(OUTPUT_DIR, f"{unique_id}_restored_faces{self.modelInUse}.zip")
             zipf_restoref = zipfile.ZipFile(zip_restoref_path, 'w', zipfile.ZIP_DEFLATED)
-            zip_cmp_path = f"output/{unique_id}_cmp{self.modelInUse}.zip"
+            zip_cmp_path = os.path.join(OUTPUT_DIR, f"{unique_id}_cmp{self.modelInUse}.zip")
             zipf_cmp = zipfile.ZipFile(zip_cmp_path, 'w', zipfile.ZIP_DEFLATED)
-            zip_restore_path = f"output/{unique_id}_restored_images{self.modelInUse}.zip"
+            zip_restore_path = os.path.join(OUTPUT_DIR, f"{unique_id}_restored_images{self.modelInUse}.zip")
             zipf_restore = zipfile.ZipFile(zip_restore_path, 'w', zipfile.ZIP_DEFLATED)
             is_auto_split_upscale = True
             name_counters = defaultdict(int)
@@ -703,13 +706,13 @@ class Upscale:
                         cropped_faces, restored_aligned, bg_upsample_img = self.face_enhancer.enhance(img_cv2, has_aligned=False, only_center_face=face_detection_only_center, paste_back=True, bg_upsample_img=bg_upsample_img, eye_dist_threshold=face_detection_threshold)
                         if cropped_faces and restored_aligned:
                             for idx, (cropped_face, restored_face) in enumerate(zip(cropped_faces, restored_aligned)):
-                                save_crop_path = f"output/{basename}_{idx:02d}_cropped_faces{self.modelInUse}.png"
+                                save_crop_path = os.path.join(OUTPUT_DIR, f"{basename}_{idx:02d}_cropped_faces{self.modelInUse}.png")
                                 self.imwriteUTF8(save_crop_path, cropped_face)
                                 zipf_cropf.write(save_crop_path, arcname=os.path.basename(save_crop_path))
-                                save_restore_path = f"output/{basename}_{idx:02d}_restored_faces{self.modelInUse}.png"
+                                save_restore_path = os.path.join(OUTPUT_DIR, f"{basename}_{idx:02d}_restored_faces{self.modelInUse}.png")
                                 self.imwriteUTF8(save_restore_path, restored_face)
                                 zipf_restoref.write(save_restore_path, arcname=os.path.basename(save_restore_path))
-                                save_cmp_path = f"output/{basename}_{idx:02d}_cmp{self.modelInUse}.png"
+                                save_cmp_path = os.path.join(OUTPUT_DIR, f"{basename}_{idx:02d}_cmp{self.modelInUse}.png")
                                 cmp_img = np.concatenate((cropped_face, restored_face), axis=1)
                                 self.imwriteUTF8(save_cmp_path, cmp_img)
                                 zipf_cmp.write(save_cmp_path, arcname=os.path.basename(save_cmp_path))
@@ -723,13 +726,13 @@ class Upscale:
                     restored_img = bg_upsample_img
                     timer.report()
                     if restored_img is None:
-                        print(f"Warning: Processing resulted in no image for '{img_path}'. Skipping output.")
-                        continue
+                       print(f"Warning: Processing resulted in no image for '{img_path}'. Skipping output.")
+                       continue
                     if save_as_png:
                         final_extension = ".png"
                     else:
                         final_extension = ".png" if img_mode == "RGBA" else (extension if extension else ".jpg")
-                    save_path = f"output/{basename}{self.modelInUse}{final_extension}"
+                    save_path = os.path.join(OUTPUT_DIR, f"{basename}{self.modelInUse}{final_extension}")
                     self.imwriteUTF8(save_path, restored_img)
                     zipf_restore.write(save_path, arcname=os.path.basename(save_path))
                     restored_img = cv2.cvtColor(restored_img, cv2.COLOR_BGR2RGB)
@@ -913,7 +916,7 @@ async def submit(
         for f in files:
             raw_bytes = await f.read()
             safe_name = os.path.basename(f.filename)
-            save_path = os.path.join('input', f"{int(time.time()*1000)}_{safe_name}")
+            save_path = os.path.join(INPUT_DIR, f"{int(time.time()*1000)}_{safe_name}")
             with open(save_path, 'wb') as out:
                 out.write(raw_bytes)
             saved_paths.append(save_path)
@@ -979,7 +982,7 @@ async def submit(
 @fastapi_app.get("/download/{filename}")
 def download(filename: str, _: bool = Depends(_verify_bearer_token)):
     safe_name = os.path.basename(filename)
-    for p in [os.path.join('output', safe_name), os.path.join('input', safe_name)]:
+    for p in [os.path.join(OUTPUT_DIR, safe_name), os.path.join(INPUT_DIR, safe_name)]:
         if os.path.isfile(p):
             return FileResponse(p, filename=safe_name)
     raise HTTPException(status_code=404, detail="File not found")