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