ananyachavan commited on
Commit
7b31438
·
verified ·
1 Parent(s): 365977d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ import whisper
7
  whisper_model = whisper.load_model("base")
8
 
9
  # Load the emotion recognition pipeline
10
- emotion_recognition = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base", return_all_scores=True)
11
 
12
  # Function to transcribe audio
13
  def transcribe_audio(audio_file):
@@ -45,4 +45,4 @@ iface = gr.Interface(
45
  )
46
 
47
  # Deploy the interface
48
- iface.launch(debug=True)
 
7
  whisper_model = whisper.load_model("base")
8
 
9
  # Load the emotion recognition pipeline
10
+ emotion_recognition = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base", top_k=None)
11
 
12
  # Function to transcribe audio
13
  def transcribe_audio(audio_file):
 
45
  )
46
 
47
  # Deploy the interface
48
+ iface.launch(share=True, debug=True)