BarBar288 commited on
Commit
fce45c7
·
verified ·
1 Parent(s): 52077bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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="facebook/detr-resnet-50")
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")