Update app.py

app.py

@@ -1,4 +1,5 @@
 import torch
+import os 
 print(f"Is CUDA available: {torch.cuda.is_available()}")
 print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
 os.system('wget https://raw.githubusercontent.com/TimDettmers/bitsandbytes/main/install_cuda.sh')
@@ -12,7 +13,7 @@ os.system('bash cuda_install.sh 117 ~/local 1')
 try:
     import detectron2
 except:
-    import os 
+    
     # os.system('pip install git+https://github.com/facebookresearch/detectron2.git')
     # os.system("python -m pip install detectron2==0.6 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html")
     os.system('git clone https://github.com/facebookresearch/detectron2.git --branch v0.6')