Spaces:

luca115
/

best_upscaling_models

Running on Zero

lucakempkes commited on 18 days ago

Commit

54005c9

1 Parent(s): 7f5aa58

update models

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,10 +6,10 @@ from image_gen_aux.utils import load_image
 from fastapi.middleware.cors import CORSMiddleware
 MODELS = {
-    "Fast: 4xNomosWebPhotoRealPLKSR": "Phips/4xNomosWebPhoto_RealPLKSR",
-    "Balanced: 4xRealESRGAN": "luca115/4xRealESRGAN",
-    "Balanced: 4xRemacri": "OzzyGT/4xRemacri",
-    "Slow: 4xSwinIRLarge": "luca115/4xSwinIRLarge",
 }
@@ -39,7 +39,7 @@ with gr.Blocks() as demo:
             model_selection = gr.Dropdown(
                 choices=list(MODELS.keys()),
-                value="Fast: 4xNomosWebPhotoRealPLKSR",
                 label="Model",
             )

 from fastapi.middleware.cors import CORSMiddleware
 MODELS = {
+    "4xNomosWebPhotoRealPLKSR": "Phips/4xNomosWebPhoto_RealPLKSR",
+    "4xRealESRGAN": "luca115/4xRealESRGAN",
+    "4xNomos8k_atd_jpg": "Phips/4xNomos8k_atd_jpg",
+    "4xSwinIRLarge": "luca115/4xSwinIRLarge",
 }
             model_selection = gr.Dropdown(
                 choices=list(MODELS.keys()),
+                value="4xSwinIRLarge",
                 label="Model",
             )