Hugging Face's logo

flaviagiammarino
/
pubmed-clip-vit-base-patch32

Zero-Shot Image Classification
Transformers
PyTorch
TensorFlow
JAX
English
clip
medical
vision
Inference Endpoints
Model card Files Community
3
flaviagiammarino commited on
Commit
1a58766
•
1 Parent(s): 7b34334

Upload 2 files

Browse files
Files changed (2) hide show
  1. scripts/pt_example.py +1 -1
  2. scripts/tf_example.py +1 -1
scripts/pt_example.py CHANGED
@@ -7,7 +7,7 @@ from transformers import CLIPProcessor, CLIPModel
7
  model = CLIPModel.from_pretrained("flaviagiammarino/pubmed-clip-vit-base-patch32")
8
  processor = CLIPProcessor.from_pretrained("flaviagiammarino/pubmed-clip-vit-base-patch32")
9
 
10
- url = "https://d168r5mdg5gtkq.cloudfront.net/medpix/img/full/synpic18755.jpg"
11
  image = Image.open(requests.get(url, stream=True).raw)
12
  text = ["Chest X-Ray", "Brain MRI", "Abdominal CT Scan"]
13
 
 
7
  model = CLIPModel.from_pretrained("flaviagiammarino/pubmed-clip-vit-base-patch32")
8
  processor = CLIPProcessor.from_pretrained("flaviagiammarino/pubmed-clip-vit-base-patch32")
9
 
10
+ url = "https://d168r5mdg5gtkq.cloudfront.net/medpix/img/full/synpic58476.jpg"
11
  image = Image.open(requests.get(url, stream=True).raw)
12
  text = ["Chest X-Ray", "Brain MRI", "Abdominal CT Scan"]
13
 
scripts/tf_example.py CHANGED
@@ -8,7 +8,7 @@ from transformers import CLIPProcessor, TFCLIPModel
8
  model = TFCLIPModel.from_pretrained("flaviagiammarino/pubmed-clip-vit-base-patch32")
9
  processor = CLIPProcessor.from_pretrained("flaviagiammarino/pubmed-clip-vit-base-patch32")
10
 
11
- url = "https://d168r5mdg5gtkq.cloudfront.net/medpix/img/full/synpic18755.jpg"
12
  image = Image.open(requests.get(url, stream=True).raw)
13
  text = ["Chest X-Ray", "Brain MRI", "Abdominal CT Scan"]
14
 
 
8
  model = TFCLIPModel.from_pretrained("flaviagiammarino/pubmed-clip-vit-base-patch32")
9
  processor = CLIPProcessor.from_pretrained("flaviagiammarino/pubmed-clip-vit-base-patch32")
10
 
11
+ url = "https://d168r5mdg5gtkq.cloudfront.net/medpix/img/full/synpic58476.jpg"
12
  image = Image.open(requests.get(url, stream=True).raw)
13
  text = ["Chest X-Ray", "Brain MRI", "Abdominal CT Scan"]
14