Spaces:
Running
on
Zero
Running
on
Zero
Update clip_analyzer.py
Browse files- clip_analyzer.py +1 -1
clip_analyzer.py
CHANGED
|
@@ -20,7 +20,7 @@ class CLIPAnalyzer:
|
|
| 20 |
Use Clip to intergrate scene understanding function
|
| 21 |
"""
|
| 22 |
|
| 23 |
-
def __init__(self, model_name: str = "ViT-
|
| 24 |
"""
|
| 25 |
εε§ε CLIP εζε¨γ
|
| 26 |
|
|
|
|
| 20 |
Use Clip to intergrate scene understanding function
|
| 21 |
"""
|
| 22 |
|
| 23 |
+
def __init__(self, model_name: str = "ViT-B/16", device: str = None):
|
| 24 |
"""
|
| 25 |
εε§ε CLIP εζε¨γ
|
| 26 |
|