Commit 0afbc0c2 authored by Aarni Koskela's avatar Aarni Koskela

Fix up `if "http" in ...:` to be more sensible startswiths

parent 89352a2f
...@@ -27,7 +27,7 @@ class UpscalerScuNET(modules.upscaler.Upscaler): ...@@ -27,7 +27,7 @@ class UpscalerScuNET(modules.upscaler.Upscaler):
scalers = [] scalers = []
add_model2 = True add_model2 = True
for file in model_paths: for file in model_paths:
if "http" in file: if file.startswith("http"):
name = self.model_name name = self.model_name
else: else:
name = modelloader.friendly_name(file) name = modelloader.friendly_name(file)
...@@ -118,7 +118,7 @@ class UpscalerScuNET(modules.upscaler.Upscaler): ...@@ -118,7 +118,7 @@ class UpscalerScuNET(modules.upscaler.Upscaler):
def load_model(self, path: str): def load_model(self, path: str):
device = devices.get_device_for('scunet') device = devices.get_device_for('scunet')
if "http" in path: if path.startswith("http"):
filename = load_file_from_url(self.model_url, model_dir=self.model_download_path, file_name=f"{self.name}.pth") filename = load_file_from_url(self.model_url, model_dir=self.model_download_path, file_name=f"{self.name}.pth")
else: else:
filename = path filename = path
......
...@@ -27,7 +27,7 @@ class UpscalerSwinIR(Upscaler): ...@@ -27,7 +27,7 @@ class UpscalerSwinIR(Upscaler):
scalers = [] scalers = []
model_files = self.find_models(ext_filter=[".pt", ".pth"]) model_files = self.find_models(ext_filter=[".pt", ".pth"])
for model in model_files: for model in model_files:
if "http" in model: if model.startswith("http"):
name = self.model_name name = self.model_name
else: else:
name = modelloader.friendly_name(model) name = modelloader.friendly_name(model)
...@@ -48,7 +48,7 @@ class UpscalerSwinIR(Upscaler): ...@@ -48,7 +48,7 @@ class UpscalerSwinIR(Upscaler):
return img return img
def load_model(self, path, scale=4): def load_model(self, path, scale=4):
if "http" in path: if path.startswith("http"):
filename = modelloader.load_file_from_url( filename = modelloader.load_file_from_url(
url=path, url=path,
model_dir=self.model_download_path, model_dir=self.model_download_path,
......
...@@ -133,7 +133,7 @@ class UpscalerESRGAN(Upscaler): ...@@ -133,7 +133,7 @@ class UpscalerESRGAN(Upscaler):
scaler_data = UpscalerData(self.model_name, self.model_url, self, 4) scaler_data = UpscalerData(self.model_name, self.model_url, self, 4)
scalers.append(scaler_data) scalers.append(scaler_data)
for file in model_paths: for file in model_paths:
if "http" in file: if file.startswith("http"):
name = self.model_name name = self.model_name
else: else:
name = modelloader.friendly_name(file) name = modelloader.friendly_name(file)
...@@ -150,7 +150,7 @@ class UpscalerESRGAN(Upscaler): ...@@ -150,7 +150,7 @@ class UpscalerESRGAN(Upscaler):
return img return img
def load_model(self, path: str): def load_model(self, path: str):
if "http" in path: if path.startswith("http"):
filename = modelloader.load_file_from_url( filename = modelloader.load_file_from_url(
url=self.model_url, url=self.model_url,
model_dir=self.model_download_path, model_dir=self.model_download_path,
......
...@@ -25,7 +25,7 @@ def gfpgann(): ...@@ -25,7 +25,7 @@ def gfpgann():
return None return None
models = modelloader.load_models(model_path, model_url, user_path, ext_filter="GFPGAN") models = modelloader.load_models(model_path, model_url, user_path, ext_filter="GFPGAN")
if len(models) == 1 and "http" in models[0]: if len(models) == 1 and models[0].startswith("http"):
model_file = models[0] model_file = models[0]
elif len(models) != 0: elif len(models) != 0:
latest_file = max(models, key=os.path.getctime) latest_file = max(models, key=os.path.getctime)
......
...@@ -82,7 +82,7 @@ def load_models(model_path: str, model_url: str = None, command_path: str = None ...@@ -82,7 +82,7 @@ def load_models(model_path: str, model_url: str = None, command_path: str = None
def friendly_name(file: str): def friendly_name(file: str):
if "http" in file: if file.startswith("http"):
file = urlparse(file).path file = urlparse(file).path
file = os.path.basename(file) file = os.path.basename(file)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment