From 2e7efe47b6c78a59f9f3d64c1d30f54751a31a56 Mon Sep 17 00:00:00 2001
From: n0kovo <n0kovo@riseup.net>
Date: Fri, 19 Jan 2024 00:39:14 +0100
Subject: Minor cleanup

---
 modules/dat_model.py | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

(limited to 'modules/dat_model.py')

diff --git a/modules/dat_model.py b/modules/dat_model.py
index 8637351c..495d5f49 100644
--- a/modules/dat_model.py
+++ b/modules/dat_model.py
@@ -1,12 +1,10 @@
 import os
-import sys
 
-from modules import modelloader, devices
+from modules import modelloader, errors
 from modules.shared import cmd_opts, opts
 from modules.upscaler import Upscaler, UpscalerData
 from modules.upscaler_utils import upscale_with_model
 
-from icecream import ic
 
 class UpscalerDAT(Upscaler):
     def __init__(self, user_path):
@@ -24,13 +22,13 @@ class UpscalerDAT(Upscaler):
             if model.name in opts.dat_enabled_models:
                 self.scalers.append(model)
 
-    def do_upscale(self, img, selected_model):
+    def do_upscale(self, img, path):
         try:
-            info = self.load_model(selected_model)
-        except Exception as e:
+            info = self.load_model(path)
+        except Exception:
             errors.report(f"Unable to load DAT model {path}", exc_info=True)
             return img
-        
+
         model_descriptor = modelloader.load_spandrel_model(
             info.local_data_path,
             device=self.device,
-- 
cgit v1.2.1