Update app.py

app.py

@@ -11,13 +11,16 @@ available_models = {
  
  
 # Create a Gradio interface with audio file and text inputs
-def classify_toxicity(audio_file, selected_model):
+def classify_toxicity(audio_file, text_file, selected_model):
     # Transcribe the audio file using Whisper ASR
-    whisper_module = evaluate.load("whisper")
-    transcription_results = whisper_module.compute(uploaded=audio_file)
- 
-    # Extract the transcribed text
-    transcribed_text = transcription_results["transcription"]
+    if audio_file != null:
+        whisper_module = evaluate.load("whisper")
+        transcription_results = whisper_module.compute(uploaded=audio_file)
+     
+        # Extract the transcribed text
+        transcribed_text = transcription_results["transcription"]
+    else:
+        transcribed_text = open(text_file, "r")
  
     # Load the selected toxicity classification model
     toxicity_module = evaluate.load("toxicity", selected_model)
@@ -30,7 +33,7 @@ def classify_toxicity(audio_file, selected_model):
 iface = gr.Interface(
     fn=classify_toxicity,
     inputs=[
-        gr.Audio(source="upload", type="filepath", label="Upload Audio File"),
+        gr.Audio(source="upload", type="filepath", label="Upload Audio File"), gr.File(label="Upload Text File"),
         gr.Radio(available_models, type="value", label="Select Model")
     ],
     outputs="text",