Update app.py
Browse files
app.py
CHANGED
@@ -130,8 +130,8 @@ def load_model(model_name):
|
|
130 |
#tokenizer = AutoTokenizer.from_pretrained(model_name)
|
131 |
tokenizer = model.tokenizer
|
132 |
|
133 |
-
if not tokenizer.pad_token:
|
134 |
-
|
135 |
|
136 |
return model, tokenizer
|
137 |
|
|
|
130 |
#tokenizer = AutoTokenizer.from_pretrained(model_name)
|
131 |
tokenizer = model.tokenizer
|
132 |
|
133 |
+
#if not tokenizer.pad_token:
|
134 |
+
# tokenizer.pad_token = "[PAD]"
|
135 |
|
136 |
return model, tokenizer
|
137 |
|