Update app.py
Browse files
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"]
|