Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr 
 import torchvision 
+from torchvision import models
 from torch import nn