Commit
•
f9cfa47
1
Parent(s):
f4909bc
Small fix
Browse files- pipeline.py +1 -0
pipeline.py
CHANGED
|
@@ -2,6 +2,7 @@
|
|
| 2 |
import io
|
| 3 |
import base64
|
| 4 |
import shutil
|
|
|
|
| 5 |
from transformers import CLIPProcessor, CLIPModel, CLIPTokenizer
|
| 6 |
|
| 7 |
|
|
|
|
| 2 |
import io
|
| 3 |
import base64
|
| 4 |
import shutil
|
| 5 |
+
import torch
|
| 6 |
from transformers import CLIPProcessor, CLIPModel, CLIPTokenizer
|
| 7 |
|
| 8 |
|