Update app.py

app.py

@@ -5,6 +5,8 @@ subprocess.run(["pip", "install", "-U", "git+https://github.com/huggingface/tran
 subprocess.run(["pip", "install", "-U", "git+https://github.com/huggingface/accelerate.git"])
 subprocess.run(["pip", "install", "datasets"])
 subprocess.run(["pip", "install", "evaluate"])
+subprocess.run(["pip", "install", "torchvision"])
+
 
 # Rest of your code
 model_checkpoint = "microsoft/resnet-50"