Hugging Face's logo

Spaces:
hopelessDev
/
Fin-senti
Sleeping

App Files Community
hopelessDev commited on
Commit
47f42b3
·
verified ·
1 Parent(s): d02c1bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -13,8 +13,10 @@ from cachetools import TTLCache, cached
13
  # -----------------------------
14
  NEWSAPI_KEY = os.environ.get("NEWSAPI_KEY", "").strip() # optional
15
  MAX_HEADLINES = 3
 
 
16
  MODEL_A = "yiyanghkust/finbert-tone"
17
- MODEL_B = "mrm8488/distilroberta-finetuned-financial-news-sentiment"
18
 
19
  # -----------------------------
20
  # Load models
@@ -31,7 +33,6 @@ LABEL_MAP = {
31
  # -----------------------------
32
  # Caching
33
  # -----------------------------
34
- # Cache up to 100 stocks, expires after 10 minutes
35
  stock_cache = TTLCache(maxsize=100, ttl=600)
36
 
37
  # -----------------------------
 
13
  # -----------------------------
14
  NEWSAPI_KEY = os.environ.get("NEWSAPI_KEY", "").strip() # optional
15
  MAX_HEADLINES = 3
16
+
17
+ # ✅ Use only public models
18
  MODEL_A = "yiyanghkust/finbert-tone"
19
+ MODEL_B = "ProsusAI/finbert"
20
 
21
  # -----------------------------
22
  # Load models
 
33
  # -----------------------------
34
  # Caching
35
  # -----------------------------
 
36
  stock_cache = TTLCache(maxsize=100, ttl=600)
37
 
38
  # -----------------------------