Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ import torch.nn as nn
|
|
6 |
import torchvision.transforms as transforms
|
7 |
# from badnet_m import BadNet
|
8 |
|
9 |
-
import timm
|
10 |
|
11 |
# model = timm.create_model("hf_hub:nateraw/resnet18-random", pretrained=True)
|
12 |
# model.train()
|
|
|
6 |
import torchvision.transforms as transforms
|
7 |
# from badnet_m import BadNet
|
8 |
|
9 |
+
# import timm
|
10 |
|
11 |
# model = timm.create_model("hf_hub:nateraw/resnet18-random", pretrained=True)
|
12 |
# model.train()
|