Update app.py

app.py

@@ -48,7 +48,12 @@ def predict_sentiment(text):
     input_ids, attention_mask = encoded_input["input_ids"], encoded_input["attention_mask"]
     with torch.no_grad():
         score = score_model(input_ids, attention_mask)[0].item()
-    return score
+    min_val, max_val = 0.3, 0.9
+    scaled_score = (score - min_val) / (max_val - min_val)
+    # Clip to ensure it stays within [0, 1] in case original score was outside [0.3, 0.9]
+    scaled_score = max(0.0, min(1.0, scaled_score))
+    
+    return scaled_score
 
 def fetch_articles(ticker):
     POLYGON_API_KEY = os.get_env('poly_api')