Spaces:

s-ahal
/

infer

Running

s-ahal commited on Apr 12

Commit

8bb6a0e

verified ·

1 Parent(s): 5f91329

Update server.py

Files changed (1) hide show

server.py CHANGED Viewed

@@ -36,8 +36,8 @@ def load_models():
     global processor, clip_model, model
     # Load CLIP model and processor
-    processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
-    clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
     # Initialize MIDM model
     input_dim = 10  # Using first 10 features as in your notebook

     global processor, clip_model, model
     # Load CLIP model and processor
+    processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch14")
+    clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch14")
     # Initialize MIDM model
     input_dim = 10  # Using first 10 features as in your notebook