update
Browse files
app.py
CHANGED
@@ -3,9 +3,9 @@ import gradio as gr
|
|
3 |
models = {
|
4 |
# "object-detection": "facebook/detr-resnet-50",
|
5 |
"image-classification": "microsoft/resnet-50",
|
6 |
-
"text-to-image": "
|
7 |
-
"image-to-text": "
|
8 |
-
"audio-classification": "
|
9 |
"audio-to-audio": "speechbrain/mtl-mimic-voicebank",
|
10 |
"automatic-speech-recognition": "jonatasgrosman/wav2vec2-large-xlsr-53-english",
|
11 |
"conversational": "microsoft/DialoGPT-medium",
|
|
|
3 |
models = {
|
4 |
# "object-detection": "facebook/detr-resnet-50",
|
5 |
"image-classification": "microsoft/resnet-50",
|
6 |
+
"text-to-image": "runwayml/stable-diffusion-v1-5",
|
7 |
+
"image-to-text": "nlpconnect/vit-gpt2-image-captioning",
|
8 |
+
"audio-classification": "mtg-upf/discogs-maest-30s-pw-73e-ts",
|
9 |
"audio-to-audio": "speechbrain/mtl-mimic-voicebank",
|
10 |
"automatic-speech-recognition": "jonatasgrosman/wav2vec2-large-xlsr-53-english",
|
11 |
"conversational": "microsoft/DialoGPT-medium",
|