Update README.md
Browse files
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`:
|