Spaces:

fffiloni
/

speech-to-image

Paused

fffiloni commited on Oct 22, 2022

Commit

9153210

1 Parent(s): 85c8597

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,8 +11,8 @@ import os
 MY_SECRET_TOKEN=os.environ.get('HF_TOKEN_SD')
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-large").to(device)
-processor = WhisperProcessor.from_pretrained("openai/whisper-large")
 diffuser_pipeline = DiffusionPipeline.from_pretrained(
     "CompVis/stable-diffusion-v1-4",

 MY_SECRET_TOKEN=os.environ.get('HF_TOKEN_SD')
 device = "cuda" if torch.cuda.is_available() else "cpu"
+model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-small").to(device)
+processor = WhisperProcessor.from_pretrained("openai/whisper-small")
 diffuser_pipeline = DiffusionPipeline.from_pretrained(
     "CompVis/stable-diffusion-v1-4",