xp3857 commited on
Commit
a53f406
·
1 Parent(s): e5867d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -67,7 +67,7 @@ def inferRestoration(img, model_name):
67
  return result
68
 
69
  def inferColorization(img):
70
- model_name == "Deoldify"
71
  model = torch.hub.load('manhkhanhad/ImageRestorationInfer', 'DeOldifyColorization')
72
  transform_list = [
73
  transforms.ToTensor(),
 
67
  return result
68
 
69
  def inferColorization(img):
70
+ model_name = "Deoldify"
71
  model = torch.hub.load('manhkhanhad/ImageRestorationInfer', 'DeOldifyColorization')
72
  transform_list = [
73
  transforms.ToTensor(),