Jr0hamza commited on
Commit
cf8e9c2
·
verified ·
1 Parent(s): f48ddf6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ label_map = {
15
  def predict_sentiment(text):
16
  result = sentiment_pipeline(text)[0]
17
  label = label_map.get(result["label"], "unknown") # Convert label
18
- return f"Sentiment: {label} (Confidence: {result['score']:.2f})"
19
 
20
  # Gradio Interface
21
  interface = gr.Interface(
 
15
  def predict_sentiment(text):
16
  result = sentiment_pipeline(text)[0]
17
  label = label_map.get(result["label"], "unknown") # Convert label
18
+ return label # Just return the label without the confidence score
19
 
20
  # Gradio Interface
21
  interface = gr.Interface(