drhead commited on
Commit
9969656
·
verified ·
1 Parent(s): 369ecce

why does your warning suggest a module that doesn't exist!

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ from torchvision.transforms import InterpolationMode
12
  import torchvision.transforms.functional as TF
13
  from huggingface_hub import hf_hub_download
14
  import numpy as np
15
- import matplotlib.colormaps as cm
16
 
17
  class Fit(torch.nn.Module):
18
  def __init__(
 
12
  import torchvision.transforms.functional as TF
13
  from huggingface_hub import hf_hub_download
14
  import numpy as np
15
+ import matplotlib.cm as cm
16
 
17
  class Fit(torch.nn.Module):
18
  def __init__(