.

app.py

@@ -62,15 +62,15 @@ def respond(
     input_ids = pipe.tokenizer.encode(prompt, return_tensors="pt")
     input_length = input_ids.shape[1]
 
+    # Ensure max_tokens is a positive integer
+    max_tokens = max(1, int(max_tokens))
+
     # Adjust max_length to be greater than input_length
     max_length = input_length + max_tokens
 
-    # Ensure max_length is an integer
-    max_length = int(max_length)
-
     outputs = pipe(
         prompt,
-        max_new_tokens=int(max_tokens),  # Ensure this is an integer
+        max_new_tokens=max_tokens,
         max_length=max_length,
         num_beams=1,
         do_sample=True,