katowtkkk commited on
Commit
53041be
·
verified ·
1 Parent(s): cd8aefc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -22,8 +22,7 @@ def audio_to_sentiment_content():
22
  sentiment_pipeline = pipeline("text-classification", model="katowtkkk/CustomModel_v2c_k")
23
  if "text" in transcription:
24
  text = transcription["text"]
25
- sentiment_prediction = sentiment_pipeline(text)
26
- st.write("prediction:", sentiment_prediction)
27
 
28
  sentiment_label = sentiment_prediction["label"]
29
  sentiment_score = sentiment_prediction["score"]
 
22
  sentiment_pipeline = pipeline("text-classification", model="katowtkkk/CustomModel_v2c_k")
23
  if "text" in transcription:
24
  text = transcription["text"]
25
+ sentiment_prediction = sentiment_pipeline(text)[0]
 
26
 
27
  sentiment_label = sentiment_prediction["label"]
28
  sentiment_score = sentiment_prediction["score"]