Update app.py

app.py

@@ -1,11 +1,16 @@
 import gradio as gr
 from optimum.intel import OVDiffusionPipeline
-import torch
+from transformers import AutoTokenizer
 from threading import Lock
 
-# Load the pipeline globally
+# Load the pipeline and tokenizer globally
 model_id = "OpenVINO/FLUX.1-schnell-int4-ov"
 pipeline = OVDiffusionPipeline.from_pretrained(model_id, device="CPU")
+
+# Explicitly load the tokenizer with fast settings
+tokenizer = AutoTokenizer.from_pretrained("t5-v1_1-xxl", use_fast=True, add_prefix_space=True)
+pipeline.text_encoder_2.tokenizer = tokenizer  # Assign the tokenizer to the T5 encoder
+
 lock = Lock()
 
 # Define the image generation function
@@ -25,6 +30,6 @@ interface = gr.Interface(
     cache_examples=False
 )
 
-# Launch the interface with queueing
+# Launch the interface
 if __name__ == "__main__":
-    interface.launch(queue=True)
+    interface.launch()  # Removed queue=True for compatibility