Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -41,11 +41,11 @@ text_to_speech_pipelines = {}
|
|
41 |
# Initialize pipelines for other tasks
|
42 |
visual_qa_pipeline = pipeline("visual-question-answering", model="dandelin/vilt-b32-finetuned-vqa")
|
43 |
document_qa_pipeline = pipeline("question-answering", model="deepset/roberta-base-squad2")
|
44 |
-
image_classification_pipeline = pipeline("image-classification", model="facebook/detr-resnet-50")
|
45 |
object_detection_pipeline = pipeline("object-detection", model="facebook/detr-resnet-50")
|
46 |
video_classification_pipeline = pipeline("video-classification", model="facebook/timesformer-base-finetuned-k400")
|
47 |
# Removed text_to_3d_pipeline as it was causing issues
|
48 |
-
keypoint_detection_pipeline = pipeline("keypoint-detection", model="
|
49 |
translation_pipeline = pipeline("translation_en_to_fr", model="Helsinki-NLP/opus-mt-en-fr")
|
50 |
summarization_pipeline = pipeline("summarization", model="facebook/bart-large-cnn")
|
51 |
text_to_audio_pipeline = pipeline("text-to-speech", model="julien-c/ljspeech_tts_train_tacotron2_raw_phn_tacotron_g2p_en_no_space")
|
|
|
41 |
# Initialize pipelines for other tasks
|
42 |
visual_qa_pipeline = pipeline("visual-question-answering", model="dandelin/vilt-b32-finetuned-vqa")
|
43 |
document_qa_pipeline = pipeline("question-answering", model="deepset/roberta-base-squad2")
|
44 |
+
image_classification_pipeline = pipeline("image-classification", model="facebook/detr-resnet-50") # This will be replaced
|
45 |
object_detection_pipeline = pipeline("object-detection", model="facebook/detr-resnet-50")
|
46 |
video_classification_pipeline = pipeline("video-classification", model="facebook/timesformer-base-finetuned-k400")
|
47 |
# Removed text_to_3d_pipeline as it was causing issues
|
48 |
+
keypoint_detection_pipeline = pipeline("keypoint-detection", model="microsoft/resnet50") # This will be replaced
|
49 |
translation_pipeline = pipeline("translation_en_to_fr", model="Helsinki-NLP/opus-mt-en-fr")
|
50 |
summarization_pipeline = pipeline("summarization", model="facebook/bart-large-cnn")
|
51 |
text_to_audio_pipeline = pipeline("text-to-speech", model="julien-c/ljspeech_tts_train_tacotron2_raw_phn_tacotron_g2p_en_no_space")
|