Update app.py

app.py

@@ -18,7 +18,7 @@ from PIL import Image
 print("cuda available:",torch.cuda.is_available())
 print("cuda device count:",torch.cuda.device_count())
 print("cuda device name:",torch.cuda.get_device_name(0))
-print(os.system("nvidia-smi"))
+# print(os.system("nvidia-smi"))
 print(os.system("nvcc --version"))
 
 empty_context = np.load("assets/contexts/empty_context.npy")

requirements.txt

@@ -1,5 +1,3 @@
-torch==1.13.1
-torchvision==0.14.1
 accelerate==0.12.0 
 absl-py 
 ml_collections 
@@ -9,5 +7,4 @@ transformers==4.23.1
 loguru
 gradio==3.34.0
 omegaconf
-wget
 xformers==0.0.16