philschmid HF staff commited on
Commit
7d4e21d
1 Parent(s): 6bacf01

Update pipeline.py

Browse files
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