Update app.py

app.py

@@ -78,6 +78,12 @@ def ensure_complete_output(output, context, max_length, temperature, top_k, top_
 
 # Text generation function for Gradio interface
 def generate_text(prompt, num_samples, max_new_tokens, temperature, top_k, top_p, repetition_penalty, eor_token_id):
+    # Add input validation
+    if num_samples is None:
+        num_samples = 1
+    elif not isinstance(num_samples, int):
+        raise ValueError("Number of samples must be an integer")
+
     with torch.no_grad():
         with ctx:
             start_ids = encode(prompt)