Hugging Face's logo

Spaces:
MaykaGR
/
proyectomod
Sleeping

App Files Community
MaykaGR commited on
Commit
3a88eeb
路
verified 路
1 Parent(s): 90fc352

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -11,6 +11,7 @@ import soundfile as sf
11
  from diffusers import StableAudioPipeline
12
  import torchsde
13
 
 
14
  processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
15
  model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large").to("cuda")
16
  pipe = StableAudioPipeline.from_pretrained("stabilityai/stable-audio-open-1.0", torch_dtype=torch.float16)
 
11
  from diffusers import StableAudioPipeline
12
  import torchsde
13
 
14
+ device = torch.device("cpu")
15
  processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
16
  model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large").to("cuda")
17
  pipe = StableAudioPipeline.from_pretrained("stabilityai/stable-audio-open-1.0", torch_dtype=torch.float16)