Charles95 commited on
Commit
9d72af7
·
verified ·
1 Parent(s): a2b6329

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
 
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",
@@ -21,7 +21,7 @@ models = {
21
  "sentence-similarity": "sentence-transformers/all-mpnet-base-v2",
22
  "text-to-speech": "facebook/mms-tts-eng",
23
  "token-classification": "benjamin/wtp-canine-s-1l",
24
- "document-question-answering": "fxmarty/tiny-doc-qa-vision-encoder-decoder",
25
  "visual-question-answering": "Salesforce/blip-vqa-capfilt-large",
26
  }
27
 
 
1
  import gradio as gr
2
 
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",
 
21
  "sentence-similarity": "sentence-transformers/all-mpnet-base-v2",
22
  "text-to-speech": "facebook/mms-tts-eng",
23
  "token-classification": "benjamin/wtp-canine-s-1l",
24
+ #"document-question-answering": "fxmarty/tiny-doc-qa-vision-encoder-decoder",
25
  "visual-question-answering": "Salesforce/blip-vqa-capfilt-large",
26
  }
27