Isgrassmann commited on
Commit
4fb21c1
·
1 Parent(s): 4fca25f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -23
app.py CHANGED
@@ -1,25 +1,4 @@
1
- import gradio as gr
2
  from transformers import pipeline
3
 
4
- # Load the Stable Diffusion TTS model
5
- tts_demo = gr.Interface.load(
6
- "models/runwayml/stable-diffusion-v1-5",
7
- title=None,
8
- description="Give me something to paint!"
9
- )
10
-
11
- # Load the text-to-image prompt generator model
12
- stt_demo = gr.Interface.load(
13
- "models/succinctly/text2image-prompt-generator",
14
- title=None,
15
- description="Let me try to guess what you're saying!"
16
- )
17
-
18
- # Load the depth estimation model
19
- depth_estimation = pipeline("depth-estimation")
20
-
21
- # Create the tabbed interface
22
- demo = gr.TabbedInterface([tts_demo, stt_demo, gr.Interface.from_pipeline(depth_estimation)], ["Paint", "Text", "Depth"])
23
-
24
- # Launch the interface
25
- demo.launch()
 
1
+ import torch
2
  from transformers import pipeline
3
 
4
+ speech_recognizer = pipeline("automatic-speech-recognition", model="facebook/wav2vec2-base-960h")