Spaces:

merve
/

compare_clip_siglip

Running on CPU Upgrade

merve HF Staff commited on Jan 9, 2024

Commit

6068e79

1 Parent(s): 8d0672c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,8 +7,8 @@ import gradio as gr
 siglip_checkpoint = "nielsr/siglip-base-patch16-224"
 clip_checkpoint = "openai/clip-vit-base-patch16"
 clip_detector = pipeline(model=clip_checkpoint, task="zero-shot-image-classification")
-siglip_model = SiglipModel.from_pretrained("nielsr/siglip-base-patch16-224")
-siglip_processor = AutoProcessor.from_pretrained("nielsr/siglip-base-patch16-224")
 def postprocess(output):

 siglip_checkpoint = "nielsr/siglip-base-patch16-224"
 clip_checkpoint = "openai/clip-vit-base-patch16"
 clip_detector = pipeline(model=clip_checkpoint, task="zero-shot-image-classification")
+siglip_model = SiglipModel.from_pretrained("google/siglip-base-patch16-224")
+siglip_processor = AutoProcessor.from_pretrained("google/siglip-base-patch16-224")
 def postprocess(output):