Hugging Face's logo

Spaces:
eaglelandsonce
/
pytorch
Running

App Files Community
eaglelandsonce commited on
Commit
600d0d3
·
verified ·
1 Parent(s): 083d57e

Update pages/19_ResNet.py

Browse files
Files changed (1) hide show
  1. pages/19_ResNet.py +1 -0
pages/19_ResNet.py CHANGED
@@ -6,6 +6,7 @@ import streamlit as st
6
  import torch
7
  import torch.nn as nn
8
  import torch.optim as optim
 
9
  from torchvision import datasets, models, transforms
10
  from torch.utils.data import DataLoader
11
  import numpy as np
 
6
  import torch
7
  import torch.nn as nn
8
  import torch.optim as optim
9
+ import torchvision # Add this import
10
  from torchvision import datasets, models, transforms
11
  from torch.utils.data import DataLoader
12
  import numpy as np