jmparejaz commited on
Commit
93390f1
1 Parent(s): 5272af9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -7
app.py CHANGED
@@ -15,7 +15,6 @@ model_nlp = AutoModelForSequenceClassification.from_pretrained("cardiffnlp/twitt
15
  model = whisper.load_model("small")
16
 
17
 
18
-
19
  def inference_audio(audio):
20
  audio = whisper.load_audio(audio)
21
  audio = whisper.pad_or_trim(audio)
@@ -27,11 +26,10 @@ def inference_audio(audio):
27
  options = whisper.DecodingOptions(fp16 = False)
28
  result = whisper.decode(model, mel, options)
29
 
30
- print(result.text)
31
- return result.text, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
32
 
33
  def inference_text(audio):
34
- text,_,_,_ =inference_audio(audio)
35
 
36
  sentiment_task = pipeline("sentiment-analysis", model=model_nlp, tokenizer=tokenizer)
37
  res=sentiment_task(text)[0]
@@ -50,6 +48,4 @@ app=gr.Interface(title="Sentiment Audio Analysis",fn=inference_text,inputs=[audi
50
 
51
 
52
 
53
- app.launch()
54
-
55
-
 
15
  model = whisper.load_model("small")
16
 
17
 
 
18
  def inference_audio(audio):
19
  audio = whisper.load_audio(audio)
20
  audio = whisper.pad_or_trim(audio)
 
26
  options = whisper.DecodingOptions(fp16 = False)
27
  result = whisper.decode(model, mel, options)
28
 
29
+ return result.text
 
30
 
31
  def inference_text(audio):
32
+ text =inference_audio(audio)
33
 
34
  sentiment_task = pipeline("sentiment-analysis", model=model_nlp, tokenizer=tokenizer)
35
  res=sentiment_task(text)[0]
 
48
 
49
 
50
 
51
+ app.launch()