fix num samples

app.py

@@ -67,6 +67,7 @@ def validate_request(request: GenerationRequest):
     
     # check that the number of samples is less than MAX_SAMPLES
     if input_dataset_info.splits[request.input_dataset_split].num_examples > MAX_SAMPLES:
+        request.num_output_examples = input_dataset_info.splits[request.input_dataset_split].num_examples
         raise Exception(f"Dataset split {request.input_dataset_split} in dataset {request.input_dataset_name} exceeds max sample limit of {MAX_SAMPLES}.")
 
     # check the prompt column exists in the dataset
@@ -134,8 +135,8 @@ def add_request_to_db(request: GenerationRequest):
             "output_dataset_token": request.output_dataset_token,
             "username": request.username,
             "email": request.email,
-            "num_output_examples": MAX_SAMPLES,  # currently always max samples
-            "private": False,
+            "num_output_examples": request.num_output_examples,
+            "private": request.private,
         }
             
         supabase.table("gen-requests").insert(data).execute()