Charles95 commited on
Commit
20b79e5
1 Parent(s): 6e7c730
Files changed (1) hide show
  1. app.py +3 -3
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": "cagliostrolab/animagine-xl-3.0",
7
- "image-to-text": "Salesforce/blip-image-captioning-large",
8
- "audio-classification": "speechbrain/emotion-recognition-wav2vec2-IEMOCAP",
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",