Spaces:

Arulkumar03
/

GroundingDINO_SOTA_Zero_Shot_Model

Sleeping

Arulkumar03 commited on Oct 17, 2023

Commit

03cec90

•

1 Parent(s): faf6b2f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ import torch
 from io import BytesIO
 from diffusers import StableDiffusionInpaintPipeline
 from huggingface_hub import hf_hub_download
 def load_model_hf(repo_id, filename, ckpt_config_filename, device='cpu'):
     cache_config_file = hf_hub_download(repo_id=repo_id, filename=ckpt_config_filename)

 from io import BytesIO
 from diffusers import StableDiffusionInpaintPipeline
 from huggingface_hub import hf_hub_download
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 def load_model_hf(repo_id, filename, ckpt_config_filename, device='cpu'):
     cache_config_file = hf_hub_download(repo_id=repo_id, filename=ckpt_config_filename)