Update app.py

app.py

@@ -69,8 +69,9 @@ def normalized(a: torch.Tensor, order=2, dim=-1):
 
 def load_clip_models(name: str = "ViT-L/14", device='cuda'):
     import open_clip
-    model2, preprocess = open_clip.create_model_and_transforms(name, device=device)
-    return model2, preprocess
+    model2, preprocess_train, preprocess_val = open_clip.create_model_and_transforms(name, device=device)
+    preprocess = preprocess_val
+    return model2, preprocess 
 
 def load_model(model_path: str, input_size=768, device: str = 'cuda', dtype=None):
     model = MLP(input_size=input_size)