Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -67,7 +67,7 @@ def inferRestoration(img, model_name):
|
|
67 |
return result
|
68 |
|
69 |
def inferColorization(img):
|
70 |
-
model_name
|
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(),
|