Update app.py
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ def inference(img, task, run_on):
|
|
43 |
|
44 |
|
45 |
parameters = {'inp_channels':3, 'out_channels':3, 'dim':48, 'num_blocks':[4,6,6,8], 'num_refinement_blocks':4, 'heads':[1,2,4,8], 'ffn_expansion_factor':2.66, 'bias':False, 'LayerNorm_type':'WithBias', 'dual_pixel_task':False}
|
46 |
-
load_arch = run_path('
|
47 |
model = load_arch['Restormer'](**parameters)
|
48 |
|
49 |
checkpoint = torch.load('deshabby.pt')
|
|
|
43 |
|
44 |
|
45 |
parameters = {'inp_channels':3, 'out_channels':3, 'dim':48, 'num_blocks':[4,6,6,8], 'num_refinement_blocks':4, 'heads':[1,2,4,8], 'ffn_expansion_factor':2.66, 'bias':False, 'LayerNorm_type':'WithBias', 'dual_pixel_task':False}
|
46 |
+
load_arch = run_path('restormer_arch.py')
|
47 |
model = load_arch['Restormer'](**parameters)
|
48 |
|
49 |
checkpoint = torch.load('deshabby.pt')
|