Hugging Face's logo

Spaces:
dxue321
/
multiple_image_recoloring
Running

App Files Community
dxue321 commited on
Commit
c776225
·
1 Parent(s): cc9e18f

run with cpu

Browse files
Files changed (11) hide show
  1. WB_sRGB/classes/__pycache__/WBsRGB.cpython-39.pyc +0 -0
  2. __pycache__/extract_palette.cpython-39.pyc +0 -0
  3. __pycache__/image.cpython-39.pyc +0 -0
  4. __pycache__/multi_image_process.cpython-39.pyc +0 -0
  5. __pycache__/recolor.cpython-39.pyc +0 -0
  6. __pycache__/solve_group_palette.cpython-39.pyc +0 -0
  7. __pycache__/utils.cpython-39.pyc +0 -0
  8. color_naming/__pycache__/colornaming.cpython-39.pyc +0 -0
  9. requirements.txt +2 -1
  10. saliency/LDF/__pycache__/infer.cpython-39.pyc +0 -0
  11. saliency/LDF/net.py +1 -1
WB_sRGB/classes/__pycache__/WBsRGB.cpython-39.pyc ADDED
Binary file (5.23 kB). View file
 
__pycache__/extract_palette.cpython-39.pyc ADDED
Binary file (2.88 kB). View file
 
__pycache__/image.cpython-39.pyc ADDED
Binary file (6.35 kB). View file
 
__pycache__/multi_image_process.cpython-39.pyc ADDED
Binary file (10.5 kB). View file
 
__pycache__/recolor.cpython-39.pyc ADDED
Binary file (2.66 kB). View file
 
__pycache__/solve_group_palette.cpython-39.pyc ADDED
Binary file (5.3 kB). View file
 
__pycache__/utils.cpython-39.pyc ADDED
Binary file (3.06 kB). View file
 
color_naming/__pycache__/colornaming.cpython-39.pyc ADDED
Binary file (2.24 kB). View file
 
requirements.txt CHANGED
@@ -8,4 +8,5 @@ scipy==1.13.1
8
  networkx==3.2.1
9
  libsvm
10
  matplotlib==3.9.4
11
- gradio==4.44.1
 
 
8
  networkx==3.2.1
9
  libsvm
10
  matplotlib==3.9.4
11
+ gradio==4.44.1
12
+ python==3.9.7
saliency/LDF/__pycache__/infer.cpython-39.pyc ADDED
Binary file (1.49 kB). View file
 
saliency/LDF/net.py CHANGED
@@ -211,6 +211,6 @@ class LDF(nn.Module):
211
 
212
  def initialize(self):
213
  if self.cfg.snapshot:
214
- self.load_state_dict(torch.load(self.cfg.snapshot))
215
  else:
216
  weight_init(self)
 
211
 
212
  def initialize(self):
213
  if self.cfg.snapshot:
214
+ self.load_state_dict(torch.load(self.cfg.snapshot, map_location=torch.device('cpu')))
215
  else:
216
  weight_init(self)