Update app.py

app.py

@@ -15,7 +15,18 @@ import random
 import time
 import requests
 import pandas as pd
-#multimodalart/flux-lora-lab - do not edit
+
+# Disable tokenizer parallelism
+os.environ["TOKENIZERS_PARALLELISM"] = "false"
+
+# Initialize the CLIP tokenizer and model
+clip_tokenizer = CLIPTokenizer.from_pretrained("openai/clip-vit-base-patch16")
+clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch16")
+clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch16")
+
+# Initialize the Longformer tokenizer and model
+longformer_tokenizer = LongformerTokenizer.from_pretrained("allenai/longformer-base-4096")
+longformer_model = LongformerModel.from_pretrained("allenai/longformer-base-4096")
 
 #Load prompts for randomization
 df = pd.read_csv('prompts.csv', header=None)
@@ -48,6 +59,15 @@ MAX_SEED = 2**32 - 1
 
 pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
 
+def process_input(input_text):
+    # Tokenize and truncate input
+    inputs = clip_processor(text=input_text, return_tensors="pt", padding=True, truncation=True, max_length=77)
+    return inputs
+
+# Example usage
+input_text = "Your long prompt goes here..."
+inputs = process_input(input_text)
+
 class calculateDuration:
     def __init__(self, activity_name=""):
         self.activity_name = activity_name