BarBar288 commited on
Commit
798fd4c
·
verified ·
1 Parent(s): 1c032b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -13
app.py CHANGED
@@ -45,7 +45,7 @@ image_classification_pipeline = pipeline("image-classification", model="facebook
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-train-256x192") # 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")
@@ -113,10 +113,6 @@ def video_classification(video):
113
  result = video_classification_pipeline(video)
114
  return result
115
 
116
- def keypoint_detection(image):
117
- result = keypoint_detection_pipeline(image)
118
- return result
119
-
120
  def translate_text(text):
121
  result = translation_pipeline(text)
122
  return result[0]["translation_text"]
@@ -200,14 +196,7 @@ with gr.Blocks() as demo:
200
  video_classification_generate.click(video_classification, inputs=video_classification_video, outputs=video_classification_output)
201
 
202
  # Removed Text-to-3D tab as it was causing issues
203
-
204
- with gr.Tab("Keypoint Detection"):
205
- keypoint_detection_image = gr.Image(label="Upload Image")
206
- keypoint_detection_generate = gr.Button("Detect Keypoints")
207
- keypoint_detection_output = gr.Image(label="Keypoint Detection Result")
208
-
209
- keypoint_detection_generate.click(keypoint_detection, inputs=keypoint_detection_image, outputs=keypoint_detection_output)
210
-
211
  with gr.Tab("Translation"):
212
  translate_text_text = gr.Textbox(label="Text")
213
  translate_text_generate = gr.Button("Translate")
 
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
+ # Removed Keypoint Detection Pipeline
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")
 
113
  result = video_classification_pipeline(video)
114
  return result
115
 
 
 
 
 
116
  def translate_text(text):
117
  result = translation_pipeline(text)
118
  return result[0]["translation_text"]
 
196
  video_classification_generate.click(video_classification, inputs=video_classification_video, outputs=video_classification_output)
197
 
198
  # Removed Text-to-3D tab as it was causing issues
199
+ # Removed Keypoint Detection tab due to issues.
 
 
 
 
 
 
 
200
  with gr.Tab("Translation"):
201
  translate_text_text = gr.Textbox(label="Text")
202
  translate_text_generate = gr.Button("Translate")