Update app.py

app.py

@@ -2,7 +2,7 @@ import os, gc
 import gradio as gr
 import numpy as np
 import random
-from transformers import AutoTokenizer, AutoFeatureExtractor
+from transformers import CLIPTokenizer, CLIPFeatureExtractor
 import spaces
 from diffusers import DiffusionPipeline, FlowMatchEulerDiscreteScheduler
 import torch
@@ -16,14 +16,12 @@ if torch.cuda.is_available():
 else:
     torch_dtype = torch.bfloat16
 
-tokenizer = AutoTokenizer.from_pretrained(
-    model_repo_id,
-    trust_remote_code=True,
+tokenizer = CLIPTokenizer.from_pretrained(
+    "openai/clip-vit-base-patch32",  # or clip-vit-large if you prefer
     use_fast=True
 )
-feature_extractor = AutoFeatureExtractor.from_pretrained(
-    model_repo_id,
-    trust_remote_code=True
+feature_extractor = CLIPFeatureExtractor.from_pretrained(
+    "openai/clip-vit-base-patch32"
 )
 
 # 3) Dispatch & load in FP16 with offloading