let5see/IOPaint
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update web_config

Browse Source
This commit is contained in:
Qing
2024-01-31 21:51:34 +08:00
parent cdac68a9a9
commit 1d2d39dfa0
8 changed files with 175 additions and 188 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
iopaint/plugins/__init__.py
View File

@@ -8,7 +8,7 @@ from .interactive_seg import InteractiveSeg
from .realesrgan import RealESRGANUpscaler
from .remove_bg import RemoveBG
from .restoreformer import RestoreFormerPlugin
from ..const import InteractiveSegModel, Device, RealESRGANModel
from ..schema import InteractiveSegModel, Device, RealESRGANModel
def build_plugins(

5
iopaint/plugins/realesrgan.py
View File

@@ -1,14 +1,11 @@
from enum import Enum
import cv2
import numpy as np
import torch
from loguru import logger
from iopaint.const import RealESRGANModel
from iopaint.helper import download_model
from iopaint.plugins.base_plugin import BasePlugin
from iopaint.schema import RunPluginRequest
from iopaint.schema import RunPluginRequest, RealESRGANModel
class RealESRGANUpscaler(BasePlugin):