update list of models

GreenWizard2015 · Apr 8, 2024 · 1428b58 · 1428b58
1 parent aefa449
commit 1428b58
Showing 1 changed file with 10 additions and 11 deletions.
diff --git a/huggingface/app.py b/huggingface/app.py
@@ -43,25 +43,24 @@ def main(args):
   models = []
   bestGroups = WBProject.groups(onlyBest=True)
   customRuns = [
+    'AR direction | ONLY SR | encoder v2, huber, RGB, residual, extra(grayscale), structured',
+    'DDPM-V | ONLY SR | encoder v2, huber, RGB, residual, extra(grayscale), structured, T start',
+    'Single-pass | ONLY SR | Basic+masking-16, encoder v2, huber, RGB, residual, extra(grayscale), structured',
+    'Single-pass | Basic+masking-16, encoder v2, huber, RGB, residual, extra(grayscale), structured',
+    'Single-pass | Basic+masking-4',
     'Single-pass | Basic+masking-8',
-    'Single-pass | Basic+masking-uniform',
+    'Single-pass | Basic+masking-16',
     'Single-pass | Basic+masking-dynamic',
-    'Single-pass | Basic',
-    'Single-pass | Basic+masking-4',
     'Single-pass | Basic+masking-32',
-    'Single-pass | Basic+masking-16',
+    'Single-pass | Basic+masking-uniform',
+    'Single-pass | Basic',
     'AR | direction',
     'AR direction, encoder v2, masking-8, residual, LAB, structured',
     'AR direction, masking-8, RGB, extra(grayscale), structured',
     'DDPM-V, encoder v2, residual, RGB, extra(grayscale), structured, T-start',
     'DDPM-V, encoder v2, residual, RGB, extra(grayscale), structured, huge',
-    'Single-pass | Basic+masking-16, encoder v2, huber, RGB, residual, extra(grayscale), structured',
-    'Huge++',
-    'DDPM-V | ONLY SR | huber, RGB, residual, extra(grayscale), structured, T start',
-    'DDPM-V | ONLY SR | encoder v2, huber, RGB, residual, extra(grayscale), structured, T start',
-    'AR direction | ONLY SR | masking-16, encoder v2, huber, RGB, residual, extra(grayscale), structured',
-    'Single-pass | ONLY SR | Basic+masking-16, encoder v2, huber, RGB, residual, extra(grayscale), structured',
     'huge++, only SR',
+    'Huge++',
   ]
   for runName in customRuns:
     models.extend( run2inference(bestGroups[runName]) )
@@ -70,7 +69,7 @@ def main(args):
   models = {model.name: model for model in models}
 
   app = AppUI(
-    processImage=infer(models),
+    MprocessImage=infer(models),
     models=models,
   )
   app.queue() # enable queueing of requests/events