Update app.py

app.py

@@ -12,6 +12,7 @@ from PIL import Image
 import numpy as np
 from small_256_model import UNet as small_UNet
 from big_1024_model import UNet as big_UNet
+from CLIP import load as load_clip
 
 # Device configuration
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
@@ -28,8 +29,11 @@ model_repo_id = "K00B404/pix2pix_flux"
 # Global model variable
 global_model = None
 
+# clip
+clip_model,clip_tokenizer = load_clip()
+
 def load_model():
-    """Load the model at startup"""
+    """Load the models at startup"""
     global global_model
     weights_name = 'big_model_weights.pth' if big else 'small_model_weights.pth'
     try:
@@ -47,6 +51,8 @@ def load_model():
         global_model = model
         return model
 
+    
+
 # Dataset class remains the same
 class Pix2PixDataset(torch.utils.data.Dataset):
     def __init__(self, ds, transform):