PyTorch upgrade + torchvision

app.py

@@ -1,15 +1,20 @@
 """
 """
 
-# Try CUDA toolkit install + AOTI base example
+# Upgrade PyTorch
+import os
+os.system('pip install --upgrade torch torchvision')
+
+# CUDA toolkit install
 from utils.cuda_toolkit import install_cuda_toolkit; install_cuda_toolkit()
+
+# Try AOTI compile + load
 import spaces
 @spaces.GPU
 def run():
     import aoti_base_example
 run()
 
-
 # Base demo
 import gradio as gr