dblasko commited on
Commit
0c55f5b
1 Parent(s): a9d5518

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -21,8 +21,8 @@ Loading the model and preprocessor:
21
  ```python
22
  from transformers import BlipForConditionalGeneration, AutoProcessor
23
 
24
- model = BlipForConditionalGeneration.from_pretrained("blip-dalle3-img2prompt").to(device)
25
- processor = AutoProcessor.from_pretrained("blip-dalle3-img2prompt")
26
  ```
27
 
28
  Inference example on an image from `laion/dalle-3-dataset`:
 
21
  ```python
22
  from transformers import BlipForConditionalGeneration, AutoProcessor
23
 
24
+ model = BlipForConditionalGeneration.from_pretrained("dblasko/blip-dalle3-img2prompt").to(device)
25
+ processor = AutoProcessor.from_pretrained("dblasko/blip-dalle3-img2prompt")
26
  ```
27
 
28
  Inference example on an image from `laion/dalle-3-dataset`: