Update app.py

app.py

@@ -26,7 +26,7 @@ summarizer = pipeline("summarization", model="human-centered-summarization/finan
 fin_model = pipeline("sentiment-analysis", model='yiyanghkust/finbert-tone', tokenizer='yiyanghkust/finbert-tone')
 fin_model_bis = pipeline("sentiment-analysis", model='ProsusAI/finbert', tokenizer='ProsusAI/finbert')
 table_to_text = pipeline('text2text-generation', model='google/flan-t5-large')
-generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')
+generator = pipeline('text-generation', model='mattshumer/Reflection-Llama-3.1-70B')
 
 def summarize_text(text):
     resp = summarizer(text)
@@ -271,10 +271,40 @@ def generate_text(df, country, theme):
     Make sure your description follows the example format and accurately reflects the data.
     """
 
-
+    prompt = 
+    f"""
+    <thinking>
+    You are given data on adverse growth for a specific country and topic. Your task is to describe the data in a clear and accurate manner, following the example provided. Reflect on how the values change from year to year, and analyze whether there is an increase or decrease compared to previous years. Consider if these changes indicate a worsening or improvement and specify if the change is strong or slight. Include the cumulative result and ensure your description aligns with the given example format.
+    </thinking>
+    
+    <output>
+    Here is an example of how to describe adverse growth data for a given country:
+    
+    Country: Australia
+    Adverse 2020: -0.43%
+    Adverse 2021: -1.99%
+    Adverse 2022: -1.20%
+    Adverse Cumulative: -3.57%
+    Topic: GDP
+    
+    Description:
+    In the adverse scenario, the GDP growth in Australia was -0.43% in 2020. It decreased further to -1.99% in 2021, showing worsening conditions. However, there was a slight improvement to -1.20% in 2022. The total cumulative adverse growth is -3.57%.
+    
+    Now, using the following data for {theme} in {country}, perform the following:
+    1. Highlight how the values change from year to year.
+    2. Describe whether the values increased or decreased compared to the previous year.
+    3. Indicate if the changes represent a worsening or improvement, and if this is strong or slight.
+    4. Include the cumulative result.
+    
+    Data:
+    {row_str}
+    Topic: {theme}
+    Make sure your description follows the example format and accurately reflects the data.
+    </output>
+"""
 
     # Generate the descriptive text using the model
-    result = table_to_text(prompt, max_length=248, temperature=0.7, top_p = 0.9, do_sample=True)[0]['generated_text']
+    result = table_to_text(prompt, max_length=248)[0]['generated_text']
     
     return result
 # Global variable