Spaces:

KrizTech100
/

Sentiment_Analysis

Sleeping

KrizTech100 commited on May 31

Commit

d98d1eb

verified ·

1 Parent(s): 5d5fe06

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -47,7 +47,7 @@ def analyze_sentiment(text):
     with torch.no_grad():
         logits = sentiment_model(**inputs).logits
     probs = F.softmax(logits, dim=-1)[0]
-    return torch.argmax(probs).item() + 1  # 1–5
 def build_segments(transcript):

     with torch.no_grad():
         logits = sentiment_model(**inputs).logits
     probs = F.softmax(logits, dim=-1)[0]
+    return torch.argmax(probs).item() + 1
 def build_segments(transcript):