Update app.py

app.py

@@ -82,13 +82,13 @@ def generate_synthetic_data(description, columns):
         formatted_prompt = format_prompt(description, columns)
         
         # Tokenize the prompt with truncation enabled
-        inputs = tokenizer_llama(formatted_prompt, return_tensors="pt", truncation=True, max_length=512).to(model_llama.device)
+        inputs = tokenizer_llama(formatted_prompt, return_tensors="pt", truncation=True, max_length=1000).to(model_llama.device)
         
         # Generate synthetic data
         with torch.no_grad():
             outputs = model_llama.generate(
                 **inputs,
-                max_length=256,
+                max_length=512,
                 top_p=generation_params["top_p"],
                 temperature=generation_params["temperature"],
                 num_return_sequences=1,