to-be commited on
Commit
67ea445
1 Parent(s): ddad5e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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('/content/drive/MyDrive/colab/denoising_dirty_documents/Restormer/basicsr/models/archs/restormer_arch.py')
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')