Update README.md
Browse files
README.md
CHANGED
|
@@ -81,6 +81,7 @@ Colab: https://colab.research.google.com/drive/1-y8HLiyS0oCj7Vpy4i7FsJ1A6kU7ROca
|
|
| 81 |
import torch
|
| 82 |
from PIL import Image
|
| 83 |
from transformers import AutoModel, AutoTokenizer, AutoProcessor
|
|
|
|
| 84 |
|
| 85 |
model_name = "5CD-AI/ColVintern-1B-v1"
|
| 86 |
|
|
|
|
| 81 |
import torch
|
| 82 |
from PIL import Image
|
| 83 |
from transformers import AutoModel, AutoTokenizer, AutoProcessor
|
| 84 |
+
import matplotlib.pyplot as plt
|
| 85 |
|
| 86 |
model_name = "5CD-AI/ColVintern-1B-v1"
|
| 87 |
|