diff --git a/bin/upscaler.py b/bin/upscaler.py index 4d92a3a..238205a 100755 --- a/bin/upscaler.py +++ b/bin/upscaler.py @@ -24,8 +24,8 @@ import shutil import tempfile import threading -MODELS_AVAILABLE = ['upconv_7_anime_style_art_rgb', 'upconv_7_photo', - 'anime_style_art_rgb', 'photo', 'anime_style_art_y'] +MODELS_AVAILABLE = ['upconv_7_anime_style_art_rgb', 'upconv_7_photo', 'anime_style_art_rgb', 'photo', + 'anime_style_art_y', 'upresnet10', 'cunet'] class Upscaler: