kassemsabeh
commited on
Commit
•
0cff8fb
1
Parent(s):
76b1db8
Fix authentication
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ from transformers import AutoTokenizer, T5ForConditionalGeneration
|
|
| 6 |
model_id = 'ksabeh/gavi'
|
| 7 |
max_input_length = 512
|
| 8 |
max_target_length = 10
|
| 9 |
-
auth_token = os.environ.get('
|
| 10 |
|
| 11 |
model = T5ForConditionalGeneration.from_pretrained(model_id, use_auth_token=auth_token)
|
| 12 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
|
|
|
| 6 |
model_id = 'ksabeh/gavi'
|
| 7 |
max_input_length = 512
|
| 8 |
max_target_length = 10
|
| 9 |
+
auth_token = os.environ.get('TOKEN')
|
| 10 |
|
| 11 |
model = T5ForConditionalGeneration.from_pretrained(model_id, use_auth_token=auth_token)
|
| 12 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|