mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 17:24:32 +02:00
add /detailers api endpoint
Signed-off-by: vladmandic <mandic00@live.com>
This commit is contained in:
@@ -102,6 +102,8 @@ def setup_model(dirname):
|
||||
gfpgan_constructor = gfpgan.GFPGANer
|
||||
|
||||
class FaceRestorerGFPGAN(modules.detailer.Detailer):
|
||||
cmd_dir = model_path
|
||||
|
||||
def name(self):
|
||||
return "GFPGAN"
|
||||
|
||||
|
||||
@@ -48,6 +48,7 @@ class YoloRestorer(Detailer):
|
||||
self.models = {} # cache loaded models
|
||||
self.list = {}
|
||||
self.ui_mode = True
|
||||
self.cmd_dir = shared.opts.yolo_dir
|
||||
self.enumerate()
|
||||
|
||||
def name(self):
|
||||
|
||||
Reference in New Issue
Block a user