mskov commited on
Commit
6bfef5d
·
1 Parent(s): 4815eb5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -14,7 +14,7 @@ available_models = {
14
 
15
 
16
  # Create a Gradio interface with audio file and text inputs
17
- def classify_toxicity(audio_file, text_file, selected_model):
18
  # Transcribe the audio file using Whisper ASR
19
  if audio_file != None:
20
  whisper_module = evaluate.load("whisper")
@@ -23,9 +23,7 @@ def classify_toxicity(audio_file, text_file, selected_model):
23
  # Extract the transcribed text
24
  transcribed_text = transcription_results["transcription"]
25
  else:
26
- with open(tmp.text_file) as f:
27
- for line in f:
28
- transcribed_text.append(line)
29
 
30
  # Load the selected toxicity classification model
31
  toxicity_module = evaluate.load("toxicity", selected_model)
@@ -38,7 +36,7 @@ def classify_toxicity(audio_file, text_file, selected_model):
38
  iface = gr.Interface(
39
  fn=classify_toxicity,
40
  inputs=[
41
- gr.Audio(source="upload", type="filepath", label="Upload Audio File"), gr.File(label="Upload Text File"),
42
  gr.Radio(available_models, type="value", label="Select Model")
43
  ],
44
  outputs="text",
 
14
 
15
 
16
  # Create a Gradio interface with audio file and text inputs
17
+ def classify_toxicity(audio_file, text_input, selected_model):
18
  # Transcribe the audio file using Whisper ASR
19
  if audio_file != None:
20
  whisper_module = evaluate.load("whisper")
 
23
  # Extract the transcribed text
24
  transcribed_text = transcription_results["transcription"]
25
  else:
26
+ transcribed_text = text_input
 
 
27
 
28
  # Load the selected toxicity classification model
29
  toxicity_module = evaluate.load("toxicity", selected_model)
 
36
  iface = gr.Interface(
37
  fn=classify_toxicity,
38
  inputs=[
39
+ gr.Audio(source="upload", type="filepath", label="Upload Audio File"), text,
40
  gr.Radio(available_models, type="value", label="Select Model")
41
  ],
42
  outputs="text",