Update app.py

app.py

@@ -44,7 +44,7 @@ document_qa_pipeline = pipeline("question-answering", model="deepset/roberta-bas
 image_classification_pipeline = pipeline("image-classification", model="facebook/detr-resnet-50")
 object_detection_pipeline = pipeline("object-detection", model="facebook/detr-resnet-50")
 video_classification_pipeline = pipeline("video-classification", model="facebook/timesformer-base-finetuned-k400")
-text_to_3d_pipeline = pipeline("text-to-3d", model="CompVis/td2s")
+# Removed text_to_3d_pipeline as it was causing issues
 keypoint_detection_pipeline = pipeline("keypoint-detection", model="facebook/detr-resnet-50")
 translation_pipeline = pipeline("translation_en_to_fr", model="Helsinki-NLP/opus-mt-en-fr")
 summarization_pipeline = pipeline("summarization", model="facebook/bart-large-cnn")
@@ -113,10 +113,6 @@ def video_classification(video):
     result = video_classification_pipeline(video)
     return result
 
-def text_to_3d(text):
-    result = text_to_3d_pipeline(text)
-    return result["image"]
-
 def keypoint_detection(image):
     result = keypoint_detection_pipeline(image)
     return result
@@ -203,12 +199,7 @@ with gr.Blocks() as demo:
         
         video_classification_generate.click(video_classification, inputs=video_classification_video, outputs=video_classification_output)
     
-    with gr.Tab("Text-to-3D"):
-        text_to_3d_text = gr.Textbox(label="Text")
-        text_to_3d_generate = gr.Button("Generate 3D")
-        text_to_3d_output = gr.Image(label="3D Model")
-        
-        text_to_3d_generate.click(text_to_3d, inputs=text_to_3d_text, outputs=text_to_3d_output)
+    # Removed Text-to-3D tab as it was causing issues
     
     with gr.Tab("Keypoint Detection"):
         keypoint_detection_image = gr.Image(label="Upload Image")
@@ -238,5 +229,12 @@ with gr.Blocks() as demo:
         
         text_to_audio_generate.click(text_to_audio, inputs=text_to_audio_text, outputs=text_to_audio_output)
 
+    with gr.Tab("Audio Classification"):
+        audio_classification_audio = gr.Audio(label="Upload Audio")
+        audio_classification_generate = gr.Button("Classify")
+        audio_classification_output = gr.Textbox(label="Classification Result")
+        
+        audio_classification_generate.click(audio_classification, inputs=audio_classification_audio, outputs=audio_classification_output)
+
 # Launch the Gradio interface
 demo.launch()