flaviagiammarino
/

pubmed-clip-vit-base-patch32

Zero-Shot Image Classification

Inference Endpoints

Model card Files Files and versions Community

flaviagiammarino commited on Jul 12, 2023

Commit

3c50594

•

1 Parent(s): 11b333d

Update scripts/flax_example.py

Files changed (1) hide show

scripts/flax_example.py +1 -1

scripts/flax_example.py CHANGED Viewed

@@ -8,7 +8,7 @@ from transformers import CLIPProcessor, FlaxCLIPModel
 model = FlaxCLIPModel.from_pretrained("flaviagiammarino/pubmed-clip-vit-base-patch32")
 processor = CLIPProcessor.from_pretrained("flaviagiammarino/pubmed-clip-vit-base-patch32")
-url = "https://d168r5mdg5gtkq.cloudfront.net/medpix/img/full/synpic9078.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
 text = ["Chest X-Ray", "Brain MRI", "Abdominal CT Scan"]

 model = FlaxCLIPModel.from_pretrained("flaviagiammarino/pubmed-clip-vit-base-patch32")
 processor = CLIPProcessor.from_pretrained("flaviagiammarino/pubmed-clip-vit-base-patch32")
+url = "https://huggingface.co/flaviagiammarino/pubmed-clip-vit-base-patch32/resolve/main/scripts/input.jpeg"
 image = Image.open(requests.get(url, stream=True).raw)
 text = ["Chest X-Ray", "Brain MRI", "Abdominal CT Scan"]