merve HF Staff commited on
Commit
0ae188f
·
verified ·
1 Parent(s): b826ad4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -41,11 +41,11 @@ from transformers import AutoProcessor, Florence2ForConditionalGeneration, BitsA
41
 
42
 
43
  model = Florence2ForConditionalGeneration.from_pretrained(
44
- "florence-community/Florence-2-large-ft",
45
  dtype=torch.bfloat16,
46
  device_map="auto",
47
  )
48
- processor = AutoProcessor.from_pretrained("florence-community/Florence-2-large-ft")
49
 
50
  url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/car.jpg?download=true"
51
  image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
 
41
 
42
 
43
  model = Florence2ForConditionalGeneration.from_pretrained(
44
+ "florence-community/Florence-2-base",
45
  dtype=torch.bfloat16,
46
  device_map="auto",
47
  )
48
+ processor = AutoProcessor.from_pretrained("florence-community/Florence-2-base")
49
 
50
  url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/car.jpg?download=true"
51
  image = Image.open(requests.get(url, stream=True).raw).convert("RGB")