gpucce commited on
Commit
daf8484
1 Parent(s): e4a3bd2
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +2 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import open_clip
4
 
5
  model, _, transform = open_clip.create_model_and_transforms(
6
  "coca_ViT-B-32",
7
- pretrained="laion2B-s13B-b90k"
8
  )
9
 
10
  def output_generate(image):
 
4
 
5
  model, _, transform = open_clip.create_model_and_transforms(
6
  "coca_ViT-B-32",
7
+ pretrained="laion2B-s13B-b90k-mscoco-2014.pt"
8
  )
9
 
10
  def output_generate(image):
requirements.txt CHANGED
@@ -1 +1,2 @@
1
- open_clip_torch
 
 
1
+ git+https://github.com/mlfoundations/open_clip
2
+ transformers