Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
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")
|