Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import gradio as gr
|
2 |
|
3 |
models = {
|
4 |
-
|
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 |
|