lxy1122 commited on
Commit
2e67624
·
1 Parent(s): e85225b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -3,6 +3,7 @@ from PIL import Image
3
  from torchvision import datasets, models, transforms
4
  import gradio as gr
5
  import os
 
6
 
7
 
8
  os.system("wget https://github.com/liuxiaoyuyuyu/vanGogh-and-Other-Artist/blob/main/artist_classes.txt")
 
3
  from torchvision import datasets, models, transforms
4
  import gradio as gr
5
  import os
6
+ import torch.nn as nn
7
 
8
 
9
  os.system("wget https://github.com/liuxiaoyuyuyu/vanGogh-and-Other-Artist/blob/main/artist_classes.txt")