jhj0517 commited on
Commit
692a471
·
1 Parent(s): 870f0f6

Update default model

Browse files
modules/image_restoration/model_downloader.py CHANGED
@@ -1,6 +1,7 @@
1
  from modules.live_portrait.model_downloader import download_model
2
 
3
  MODELS_REALESRGAN_URL = {
 
4
  "realesr-general-x4v3": "https://huggingface.co/jhj0517/realesr-general-x4v3/resolve/main/realesr-general-x4v3.pth",
5
  }
6
 
 
1
  from modules.live_portrait.model_downloader import download_model
2
 
3
  MODELS_REALESRGAN_URL = {
4
+ "RealESRGAN_x2": "https://huggingface.co/ai-forever/Real-ESRGAN/resolve/main/RealESRGAN_x2.pth",
5
  "realesr-general-x4v3": "https://huggingface.co/jhj0517/realesr-general-x4v3/resolve/main/realesr-general-x4v3.pth",
6
  }
7
 
modules/image_restoration/real_esrgan_inferencer.py CHANGED
@@ -23,7 +23,7 @@ class RealESRGANInferencer:
23
 
24
  def load_model(self,
25
  model_name: Optional[str] = None,
26
- scale: int = 1,
27
  progress: gr.Progress = gr.Progress()):
28
  if model_name is None:
29
  model_name = self.default_model
 
23
 
24
  def load_model(self,
25
  model_name: Optional[str] = None,
26
+ scale: int = 2,
27
  progress: gr.Progress = gr.Progress()):
28
  if model_name is None:
29
  model_name = self.default_model