Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import os
-os.system('pip3 install torch  -q')
+os.system('pip3 install torch torchvision -q')
 import torch
 import sys
 import numpy as np