florentgbelidji HF staff philschmid HF staff commited on
Commit
f3126f3
1 Parent(s): 6bacf01

Update pipeline.py (#2)

Browse files

- Update pipeline.py (7d4e21d910bb3467711970013f28ca64ce919334)


Co-authored-by: Philipp Schmid <[email protected]>

Files changed (1) hide show
  1. pipeline.py +1 -0
pipeline.py CHANGED
@@ -3,6 +3,7 @@ from PIL import Image
3
  import requests
4
  import torch
5
  import base64
 
6
  from io import BytesIO
7
  from blip import blip_decoder
8
  from torchvision import transforms
 
3
  import requests
4
  import torch
5
  import base64
6
+ import os
7
  from io import BytesIO
8
  from blip import blip_decoder
9
  from torchvision import transforms