khang119966 commited on
Commit
59c925a
1 Parent(s): 1fd1886

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
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