Update README.md
Browse files
README.md
CHANGED
@@ -39,7 +39,7 @@ To start using this model, run the following:
|
|
39 |
```python
|
40 |
from transformers import AutoProcessor, PaliGemmaForConditionalGeneration
|
41 |
|
42 |
-
model = PaliGemmaForConditionalGeneration.from_pretrained("agentsea/paligemma-3b-ft-widgetcap-waveui-448")
|
43 |
processor = AutoProcessor.from_pretrained("google/paligemma-3b-pt-448")
|
44 |
```
|
45 |
|
|
|
39 |
```python
|
40 |
from transformers import AutoProcessor, PaliGemmaForConditionalGeneration
|
41 |
|
42 |
+
model = PaliGemmaForConditionalGeneration.from_pretrained("agentsea/paligemma-3b-ft-widgetcap-waveui-448").eval()
|
43 |
processor = AutoProcessor.from_pretrained("google/paligemma-3b-pt-448")
|
44 |
```
|
45 |
|