akhaliq HF staff commited on
Commit
c65c38e
β€’
1 Parent(s): 03423aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -4
app.py CHANGED
@@ -26,10 +26,6 @@ import imageio
26
  from PIL import ImageFile, Image
27
  ImageFile.LOAD_TRUNCATED_IMAGES = True
28
  import gradio as gr
29
- import nvidia_smi
30
- nvidia_smi.nvmlInit()
31
- handle = nvidia_smi.nvmlDeviceGetHandleByIndex(0)
32
- # card id 0 hardcoded here, there is also a call to get all available card ids, so we could iterate
33
  torch.hub.download_url_to_file('https://images.pexels.com/photos/158028/bellingrath-gardens-alabama-landscape-scenic-158028.jpeg', 'garden.jpeg')
34
  torch.hub.download_url_to_file('https://images.pexels.com/photos/68767/divers-underwater-ocean-swim-68767.jpeg', 'coralreef.jpeg')
35
  torch.hub.download_url_to_file('https://images.pexels.com/photos/803975/pexels-photo-803975.jpeg', 'cabin.jpeg')
 
26
  from PIL import ImageFile, Image
27
  ImageFile.LOAD_TRUNCATED_IMAGES = True
28
  import gradio as gr
 
 
 
 
29
  torch.hub.download_url_to_file('https://images.pexels.com/photos/158028/bellingrath-gardens-alabama-landscape-scenic-158028.jpeg', 'garden.jpeg')
30
  torch.hub.download_url_to_file('https://images.pexels.com/photos/68767/divers-underwater-ocean-swim-68767.jpeg', 'coralreef.jpeg')
31
  torch.hub.download_url_to_file('https://images.pexels.com/photos/803975/pexels-photo-803975.jpeg', 'cabin.jpeg')