Update app.py

app.py

@@ -36,14 +36,22 @@ except ImportError as e:
 # }
 
 MODELS = {
-    #"Kaloscope v2.0": {
-    #    "type": "pytorch",
-    #    "path": "",
-    #    "repo_id": "heathcliff01/Kaloscope2.0",
-    #    "subfolder": "448-90.13",
-    #    "filename": "best_checkpoint.pth",
-    #    "arch": "lsnet_xl_artist_448",
-    #},
+    "Kaloscope v2.0 ONNX": {
+        "type": "onnx",
+        "path": "",
+        "repo_id": "DraconicDragon/Kaloscope-onnx",
+        "subfolder": "v2.0",
+        "filename": "kaloscope_2-0.onnx",
+        "arch": "lsnet_xl_artist_448",
+    },
+    "Kaloscope v2.0": {
+        "type": "pytorch",
+        "path": "",
+        "repo_id": "heathcliff01/Kaloscope2.0",
+        "subfolder": "448-90.13",
+        "filename": "best_checkpoint.pth",
+        "arch": "lsnet_xl_artist_448",
+    },
     "Kaloscope v1.1 ONNX": {
         "type": "onnx",
         "path": "",
@@ -177,7 +185,7 @@ def get_model_inference(model_name):
             # Prefer explicit csv repo if provided, otherwise fall back to the model's repo_id
             csv_repo = config.get("csv_repo_id") or config.get("repo_id")
             csv_filename = config.get("csv_filename") or CSV_FILENAME
-            csv_subfolder = config.get("csv_subfolder")
+            csv_subfolder = config.get("csv_subfolder") or config.get("subfolder")
 
             if not csv_repo:
                 raise gr.Error(f"No repo available to find class mapping CSV for model '{model_name}'")