Update app.py

app.py

@@ -188,12 +188,6 @@ def invert_displayed_image(image):
     return inverted
 
 
-MAX_TOKENS = 150
-
-def count_tokens(text):
-    """Count the number of tokens in the text."""
-    return len(text.split())
-
 @spaces.GPU()
 def inference(
     qr_code_content: str,
@@ -228,13 +222,6 @@ def inference(
         if qrcode_image is None and qr_code_content == "":
             raise gr.Error("QR Code Image or QR Code Content is required")
 
-        # Validate the length of the prompt
-        if count_tokens(prompt) > MAX_TOKENS:
-            raise gr.Error(f"Prompt exceeds the maximum allowed tokens of {MAX_TOKENS}")
-
-        if count_tokens(negative_prompt) > MAX_TOKENS:
-            raise gr.Error(f"Negative prompt exceeds the maximum allowed tokens of {MAX_TOKENS}")
-
         pipe.scheduler = SAMPLER_MAP[sampler](pipe.scheduler.config)
 
         if seed == -1:
@@ -307,29 +294,6 @@ def inference(
         print(f"Unexpected error in inference: {str(e)}")
         return Image.new('RGB', (1024, 1024), color='white'), -1
 
-
-
-def split_prompt(prompt, max_length=77):
-    """Split the prompt into chunks that do not exceed the max_length."""
-    words = prompt.split()
-    chunks = []
-    current_chunk = []
-    current_length = 0
-
-    for word in words:
-        if current_length + len(word) + 1 > max_length:
-            chunks.append(" ".join(current_chunk))
-            current_chunk = [word]
-            current_length = len(word) + 1
-        else:
-            current_chunk.append(word)
-            current_length += len(word) + 1
-
-    if current_chunk:
-        chunks.append(" ".join(current_chunk))
-
-    return chunks
-
 def invert_init_image_display(image):
     if image is None:
         return None