Update app.py

app.py

@@ -36,7 +36,7 @@ db = get_db_firestore()
 asr = pipeline("automatic-speech-recognition", "facebook/wav2vec2-base-960h")
 
 # create Text Classification pipeline
-classifier = pipeline("text-classification")
+#classifier = pipeline("text-classification")
 
 # create text generator pipeline
 story_gen = pipeline("text-generation", "pranavpsv/gpt2-genre-story-generator")
@@ -148,14 +148,13 @@ with demo:
     audio = gr.Audio(label="Output", interactive=False)
     
     b1 = gr.Button("Recognize Speech")
-    b2 = gr.Button("Classify Sentiment")
-    b3 = gr.Button("Save Speech to Text")
-    b4 = gr.Button("Retrieve All")
-
-    b1.click(speech_to_text, inputs=audio_file, outputs=text)
-    b2.click(text_to_sentiment, inputs=text, outputs=label)
-    b3.click(upsert, inputs=text, outputs=saved)
-    b4.click(selectall, inputs=text, outputs=savedAll)
+#    b2 = gr.Button("Classify Sentiment")
+#    b3 = gr.Button("Save Speech to Text")
+#    b4 = gr.Button("Retrieve All")
+
+#    b2.click(text_to_sentiment, inputs=text, outputs=label)
+ #   b3.click(upsert, inputs=text, outputs=saved)
+ #   b4.click(selectall, inputs=text, outputs=savedAll)
     
     input_story_type = gr.Radio(choices=['superhero', 'action', 'drama', 'horror', 'thriller', 'sci_fi'], value='sci_fi', label="Genre")
     input_start_text = gr.Textbox(placeholder='A teddy bear outer space', label="Starting Text")
@@ -170,6 +169,9 @@ with demo:
     output_interpolation = gr.Video(label="Generated Video")
             
     # Bind functions to buttons
+
+    b1.click(speech_to_text, inputs=audio_file, outputs=input_start_text )
+    
     button_gen_story.click(fn=generate_story, inputs=[input_story_type , input_start_text], outputs=output_generated_story)
     button_gen_images.click(fn=generate_images, inputs=output_generated_story, outputs=output_gallery)
     button_gen_video.click(fn=generate_interpolation, inputs=output_gallery, outputs=output_interpolation)