Update app.py

app.py

@@ -11,14 +11,19 @@ from image_gen_aux.utils import load_image
 
 # My models
 MODELS = {
-    "4xNomos2_hq_drct-l": "Phips/4xNomos2_hq_drct-l",
     "4xNomosWebPhoto_RealPLKSR": "Phips/4xNomosWebPhoto_RealPLKSR",
+    "4xNomos2_hq_dat2": "Phips/4xNomos2_hq_dat2",
+    "4xNomos2_hq_drct-l": "Phips/4xNomos2_hq_drct-l",
+    "4xNomos2_hq_atd": "Phips/4xNomos2_hq_atd",
     "4xRealWebPhoto_v4_dat2": "Phips/4xRealWebPhoto_v4_dat2",
     "4xRealWebPhoto_v3_atd": "Phips/4xRealWebPhoto_v3_atd",
     "4xNomos8k_atd_jpg": "Phips/4xNomos8k_atd_jpg",
     "4xNomosUni_rgt_multijpg": "Phips/4xNomosUni_rgt_multijpg",
     "4xLSDIRDAT": "Phips/4xLSDIRDAT",
     "4xSSDIRDAT": "Phips/4xSSDIRDAT",
+    "4xNomos2_otf_esrgan": "Phips/4xNomos2_otf_esrgan",
+    "4xNomosWebPhoto_esrgan": "Phips/4xNomosWebPhoto_esrgan",
+    "4xNomosWebPhoto_atd": "Phips/4xNomosWebPhoto_atd",
     "4xNomos8kHAT-L_otf": "Phips/4xNomos8kHAT-L_otf",
     "4xNomosUniDAT_otf": "Phips/4xNomosUniDAT_otf",
     "4xNomosUniDAT_bokeh_jpg": "Phips/4xNomosUniDAT_bokeh_jpg",
@@ -70,7 +75,7 @@ with gr.Blocks() as demo:
 
             model_selection = gr.Dropdown(
                 choices=list(MODELS.keys()),
-                value="4xNomos2_hq_drct-l",
+                value="4xNomosWebPhoto_RealPLKSR",
                 label="Model",
             )