Update app.py

app.py

@@ -13,8 +13,9 @@ from image_gen_aux.utils import load_image
 # I simply want them to message me first so I know that everythings okay with having their model as a selection here since they are the author of that model. If they want their model on here or not basically.
 # I load models from huggingface model cards though, so the model should be hosted on huggingface.
 MODELS = {
-    "RealPLKSR X4": "OzzyGT/4xNomosWebPhoto_RealPLKSR",
-    "DAT-2 RealWebPhoto X4": "Phips/4xRealWebPhoto_v4_dat2",
+    "4xNomosWebPhoto_RealPLKSR": "OzzyGT/4xNomosWebPhoto_RealPLKSR",
+    "4xRealWebPhoto_v4_dat2": "Phips/4xRealWebPhoto_v4_dat2",
+    "4xNomos2_hq_drct-l": "Phips/4xNomos2_hq_drct-l",
 }
 
 
@@ -45,7 +46,7 @@ with gr.Blocks() as demo:
 
             model_selection = gr.Dropdown(
                 choices=list(MODELS.keys()),
-                value="RealPLKSR X4",
+                value="4xNomos2_hq_drct-l",
                 label="Model",
             )