Hugging Face's logo

Spaces:
MaykaGR
/
proyectomod
Sleeping

App Files Community
MaykaGR commited on
Commit
7a61423
路
verified 路
1 Parent(s): 9c6458f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -6,6 +6,8 @@ from transformers import BlipProcessor, BlipForConditionalGeneration
6
  import torch
7
  import soundfile as sf
8
  from diffusers import StableAudioPipeline
 
 
9
 
10
  device = torch.device("cpu")
11
  processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
 
6
  import torch
7
  import soundfile as sf
8
  from diffusers import StableAudioPipeline
9
+ import os
10
+ os.environ['CUDA_VISIBLE_DEVICES'] = ''
11
 
12
  device = torch.device("cpu")
13
  processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")