Spaces:

deedax
/

face-facts

Sleeping

deedax commited on May 15, 2023

Commit

a6b83fe

1 Parent(s): 79c4062

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -2,6 +2,7 @@ import torch.nn as nn
 import torch
 import pytorch_lightning as pl
 import torch.nn.functional as F
 class CustomModelMain(nn.Module):
     def __init__(self, problem_type, n_classes):
         super().__init__()

 import torch
 import pytorch_lightning as pl
 import torch.nn.functional as F
+import timm
 class CustomModelMain(nn.Module):
     def __init__(self, problem_type, n_classes):
         super().__init__()