Update app.py

app.py

@@ -207,7 +207,6 @@ def generate_text(df, country, theme):
         # Create a formatted string with colons and percentages
         formatted_row = []
         for col, value in row.items():
-            print(col)
             if col != 'Country':  # Exclude 'Country' or format differently if needed
                 if isinstance(value, (int, float)):  # Add percentage sign for numeric values
                     value_str = f"{value:.6f}%"
@@ -221,7 +220,7 @@ def generate_text(df, country, theme):
     row = df[df['Country'] == country].iloc[0]
     row_str = format_row_for_prompt(row)
     #row_str = row.to_string(index=True)
-    
+    print(row_str)
     simple_prompt = f"""
     Here is the data for {theme} in {country}:
     {row_str}
@@ -274,7 +273,7 @@ def generate_text(df, country, theme):
 
 
     # Generate the descriptive text using the model
-    result = table_to_text(prompt, max_length=200, temperature = 0.7, top_p = 0.9)[0]['generated_text']
+    result = table_to_text(prompt, max_length=200, temperature = 0.7, top_p = 0.9, do_sample=True)[0]['generated_text']
     
     return result
 # Global variable