Update app.py

app.py

@@ -1,3 +1,4 @@
+import torchvision
 import gradio as gr
 import os
 import random