Update app.py

app.py

@@ -163,6 +163,11 @@ def sample_news(news, k=5):
     return [news[i] for i in sorted(random.sample(range(len(news)), k))]
 
 
+def latest_news(news, k=10):
+    # Sort news by date in descending order and select the latest k items
+    sorted_news = sorted(news, key=lambda x: x['date'], reverse=True)
+    return sorted_news[:k]
+    
 def get_current_basics(symbol, curday):
 
     basic_financials = finnhub_client.company_basic_financials(symbol, 'all')
@@ -201,7 +206,7 @@ def get_all_prompts_online(symbol, data, curday, with_basics=True):
     prompt = ""
     for i in range(-len(prev_rows), 0):
         prompt += "\n" + prev_rows[i][0]
-        sampled_news = sample_news(
+        sampled_news = latest_news(
             prev_rows[i][1],
             min(5, len(prev_rows[i][1]))
         )
@@ -260,7 +265,7 @@ def predict(ticker, date, n_weeks, use_basics):
     print("Inputs loaded onto devices.")
         
     res = model.generate(
-        **inputs, max_length=4096, do_sample=True,
+        **inputs, max_length=4096, do_sample=False,
         eos_token_id=tokenizer.eos_token_id,
         use_cache=True, streamer=streamer
     )