Update app.py

app.py

@@ -246,6 +246,24 @@ def generate_text(df, country, theme):
     
     Describe, using the similar pattern from the example, the changes for the provided country and years. Highlight how the values change year by year and whether they increased or decreased. Do not mention any other countries or years, and describe exactly what is in the table. Keep the description simple and direct.
     """
+    prompt = f"""
+    Here is an example of how to describe adverse growth data for a given country:
+    Country: Australia
+    Adverse 1990: -0.43%
+    Adverse 1991: -1.99%
+    Adverse 1992: -1.20%
+    Adverse Cumulative: -3.57%
+    Topic: GDP
+    
+    Description:
+    In the adverse scenario, the GDP growth in Australia was -0.43% in 1990. It worsened to -1.99% in 1991 and slightly improved to -1.20% in 1992. The total cumulative adverse growth was -3.57%.
+    
+    Now, using the following data for {theme} in {country}, describe the adverse growth:
+    {row_str}
+    Topic: {theme}
+    
+    Your task is to describe the changes in the provided data year by year, just like in the example. Focus only on the provided country and values, and do not introduce any new countries or years. Ensure the description follows the pattern in the example, highlighting whether the values increased or decreased each year.
+    """
     prompt1 = f"""
     Here is an example of how to describe adverse growth data for a given country:
     
@@ -272,7 +290,7 @@ def generate_text(df, country, theme):
     """
 
     # Generate the descriptive text using the model
-    result = table_to_text(prompt, max_length=240)[0]['generated_text']
+    result = table_to_text(prompt, max_length=240, temperature = 0.7, do_sample = True)[0]['generated_text']
     
     return result
 # Global variable