Update app.py

app.py

@@ -4,13 +4,14 @@ import random
 #import spaces #[uncomment to use ZeroGPU]
 from diffusers import DiffusionPipeline, FluxPipeline
 import torch
-import transformers
-
-transformers.utils.move_cache()
+from transformers import AutoTokenizer
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_repo_id = "black-forest-labs/FLUX.1-schnell" #Replace to the model you would like to use
 
+tokenizer = AutoTokenizer.from_pretrained(model_repo_id, use_fast=True)
+tokenizer.add_prefix_space = False 
+
 if torch.cuda.is_available():
     torch_dtype = torch.float16
 else: