meraj12 commited on
Commit
c2aeaad
·
verified ·
1 Parent(s): ac942b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -15,7 +15,8 @@ client = Groq(api_key="your_api_key_here")
15
  device = "cpu"
16
  whisper_model = AutoModelForSpeechSeq2Seq.from_pretrained("openai/whisper-tiny").to(device)
17
  processor = AutoProcessor.from_pretrained("openai/whisper-tiny")
18
- whisper_pipeline = pipeline("automatic-speech-recognition", model=whisper_model, processor=processor, device=device)
 
19
 
20
  # Load Stable Diffusion model
21
  sd_model = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5").to(device)
 
15
  device = "cpu"
16
  whisper_model = AutoModelForSpeechSeq2Seq.from_pretrained("openai/whisper-tiny").to(device)
17
  processor = AutoProcessor.from_pretrained("openai/whisper-tiny")
18
+ whisper_pipeline = pipeline("automatic-speech-recognition", model="openai/whisper-tiny", device=device)
19
+
20
 
21
  # Load Stable Diffusion model
22
  sd_model = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5").to(device)