add requirements

App_main.py

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

requirements.txt

@@ -0,0 +1,8 @@
+tensorboardX
+matplotlib
+einops
+yaml
+opencv-python
+pandas
+scikit-image
+tqdm