Update app.py

app.py

@@ -11,7 +11,7 @@ available_models = {
  
  
 # Create a Gradio interface with audio file and text inputs
-def classify_toxicity(audio_file, text_input, selected_model):
+def classify_toxicity(audio_file, selected_model):
     # Transcribe the audio file using Whisper ASR
     whisper_module = evaluate.load("whisper")
     transcription_results = whisper_module.compute(uploaded=audio_file)
@@ -25,13 +25,12 @@ def classify_toxicity(audio_file, text_input, selected_model):
  
     toxicity_score = toxicity_results["toxicity"][0]
  
-    return f"Toxicity Score ({available_models[selected_model]}): {toxicity_score:.4f}"
+    return transcribed_text, f"Toxicity Score ({available_models[selected_model]}): {toxicity_score:.4f}"
  
 iface = gr.Interface(
     fn=classify_toxicity,
     inputs=[
-        gr.Audio(source="upload", type="filepath", label="Upload Audio File (Optional)"),
-        "text",
+        gr.Audio(source="upload", type="filepath", label="Upload Audio File"),
         gr.Radio(available_models, type="value", label="Select Model")
     ],
     outputs="text",