add

App_main.py

@@ -5,7 +5,7 @@ import gradio as gr
 
 os.system('nvidia-smi')
 os.system('ls /usr/local')
-# os.system('pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu113')
+os.system('pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu116')
 import torch
 from torchvision import transforms
 from torchvision.transforms import InterpolationMode