DimaML commited on
Commit
a7d5366
·
verified ·
1 Parent(s): 343d11b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  import numpy as np
3
  import torch
4
  import torch.nn as nn
5
- from torchvision import transforms, dataset, models
6
 
7
  transformer = models.ResNet18_Weights.IMAGENET1K_V1.transforms()
8
 
 
2
  import numpy as np
3
  import torch
4
  import torch.nn as nn
5
+ from torchvision import transforms, datasets, models
6
 
7
  transformer = models.ResNet18_Weights.IMAGENET1K_V1.transforms()
8