Update app.py

app.py

@@ -82,7 +82,8 @@ 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=1000).to(model_llama.device)
+        inputs = tokenizer_llama(formatted_prompt, return_tensors="pt", truncation=True, max_length=512).to(model_llama.device)
+        print(f"Input Tensor Size: {inputs['input_ids'].size()}")
         
         # Generate synthetic data
         with torch.no_grad():