Update app.py

app.py

@@ -1,5 +1,4 @@
 import os
-os.system("nvidia-smi")
 os.system("pip install --upgrade torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html")
 os.system("git clone https://github.com/openai/CLIP")
 os.system("pip install -e ./CLIP")
@@ -22,7 +21,7 @@ from tqdm.notebook import tqdm
 from torchvision.transforms import Compose, Resize, ToTensor, Normalize
 from einops import rearrange
 import gradio as gr
-
+print(torch.cuda.get_device_name(0))
 device = torch.device('cuda:0')
 def fetch(url_or_path):
     if str(url_or_path).startswith('http://') or str(url_or_path).startswith('https://'):