gsarti commited on
Commit
3a1052f
·
1 Parent(s): 7d34870

Add HF token secret

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import re
 
2
  import spaces
3
  import gradio as gr
4
  from transformers import AutoTokenizer, AutoModelForCausalLM
@@ -18,7 +19,8 @@ Chiave di lettura: {key}<|end|>
18
  eureka5_test_data = load_dataset(
19
  'gsarti/eureka-rebus', 'llm_sft',
20
  data_files=["id_test.jsonl", "ood_test.jsonl"],
21
- split = "train"
 
22
  )
23
 
24
  OUTPUTS_BASE_URL = "https://raw.githubusercontent.com/gsarti/verbalized-rebus/main/outputs/"
 
1
  import re
2
+ import os
3
  import spaces
4
  import gradio as gr
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
 
19
  eureka5_test_data = load_dataset(
20
  'gsarti/eureka-rebus', 'llm_sft',
21
  data_files=["id_test.jsonl", "ood_test.jsonl"],
22
+ split = "train",
23
+ token=os.environ["HF_TOKEN"]
24
  )
25
 
26
  OUTPUTS_BASE_URL = "https://raw.githubusercontent.com/gsarti/verbalized-rebus/main/outputs/"