paugar commited on
Commit
7118776
·
verified ·
1 Parent(s): cf7aa96

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,10 +1,11 @@
 
1
  import gradio as gr
2
  from PIL import Image
3
  import os
4
  import torch
5
  import torch.nn.functional as F
6
  import torchvision.transforms as transforms
7
- import torchvision
8
  from torchkeras import plots
9
  import numpy as np
10
  import yaml
 
1
+ %pip install torchkeras
2
  import gradio as gr
3
  from PIL import Image
4
  import os
5
  import torch
6
  import torch.nn.functional as F
7
  import torchvision.transforms as transforms
8
+ import torchvisiona
9
  from torchkeras import plots
10
  import numpy as np
11
  import yaml