Kathirsci commited on
Commit
bfbf6b1
1 Parent(s): fe94180

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -1
app.py CHANGED
@@ -28,7 +28,22 @@ if not all([NEWSAPI_KEY, APCA_API_KEY_ID, APCA_API_SECRET_KEY]):
28
 
29
  # Initialize NewsAPI client
30
  newsapi = NewsApiClient(api_key=NEWSAPI_KEY)
31
-
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
32
  # Initialize Alpaca Trade API client
33
  alpaca_api = tradeapi.REST(APCA_API_KEY_ID, APCA_API_SECRET_KEY, base_url='https://paper-api.alpaca.markets')
34
 
 
28
 
29
  # Initialize NewsAPI client
30
  newsapi = NewsApiClient(api_key=NEWSAPI_KEY)
31
+ # Initialize the chatbot
32
+ HF_TOKEN = os.getenv("HF_TOKEN")
33
+ llm = HuggingFaceEndpoint(
34
+ #repo_id="google/gemma-2b-it",
35
+ #repo_id="abhinand/gemma-2b-tamil",
36
+ repo_id="mrm8488/distilroberta-finetuned-financial-news-sentiment-analysis",
37
+ #repo_id="google/gemma-2b",
38
+
39
+ task="financial-analysis",
40
+ max_new_tokens=512,
41
+ top_k=5,
42
+ temperature=0.2,
43
+ repetition_penalty=1.03,
44
+ timeout=300,
45
+ huggingfacehub_api_token=HF_TOKEN
46
+ )
47
  # Initialize Alpaca Trade API client
48
  alpaca_api = tradeapi.REST(APCA_API_KEY_ID, APCA_API_SECRET_KEY, base_url='https://paper-api.alpaca.markets')
49