Spaces:
Running
Running
Update models/tag2text.py
Browse files- models/tag2text.py +1 -1
models/tag2text.py
CHANGED
|
@@ -26,7 +26,7 @@ def read_json(rpath):
|
|
| 26 |
with open(rpath, 'r') as f:
|
| 27 |
return json.load(f)
|
| 28 |
|
| 29 |
-
delete_tag_index = [127, 3351, 3265, 3338, 3355, 3359]
|
| 30 |
|
| 31 |
class Tag2Text_Caption(nn.Module):
|
| 32 |
def __init__(self,
|
|
|
|
| 26 |
with open(rpath, 'r') as f:
|
| 27 |
return json.load(f)
|
| 28 |
|
| 29 |
+
delete_tag_index = [127,2961, 3351, 3265, 3338, 3355, 3359]
|
| 30 |
|
| 31 |
class Tag2Text_Caption(nn.Module):
|
| 32 |
def __init__(self,
|