why does your warning suggest a module that doesn't exist!
Browse files
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.
|
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__(
|