Hugging Face's logo

Spaces:
Ayemos
/
highlight_text_based_on_eye_movement
Runtime error

App Files Community
Ayemos commited on
Commit
f8a8921
•
1 Parent(s): 3dcdd1a

fix typo

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,8 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer
7
 
8
  device = torch.device("cuda:0") if torch.cuda.is_available() else torch.device("cpu")
9
  # dundee-geco
10
- tokenizer = GPT2Tokenizer.from_pretrained("dendee-geco_test-on-zuco1.0_gpt2_tmptoken_TRT_bs32_lr1e6_linearLR")
11
- model = GPT2LMHeadModel.from_pretrained("dendee-geco_test-on-zuco1.0_gpt2_tmptoken_TRT_bs32_lr1e6_linearLR", return_dict=True)
12
 
13
  # zuco
14
  tokenizer = GPT2Tokenizer.from_pretrained("zuco1.0_gpt2_tmptoken_TRT_bs32_lr1e5_linearLR")
 
7
 
8
  device = torch.device("cuda:0") if torch.cuda.is_available() else torch.device("cpu")
9
  # dundee-geco
10
+ # tokenizer = GPT2Tokenizer.from_pretrained("dendee-geco_test-on-zuco1.0_gpt2_tmptoken_TRT_bs32_lr1e6_linearLR")
11
+ # model = GPT2LMHeadModel.from_pretrained("dendee-geco_test-on-zuco1.0_gpt2_tmptoken_TRT_bs32_lr1e6_linearLR", return_dict=True)
12
 
13
  # zuco
14
  tokenizer = GPT2Tokenizer.from_pretrained("zuco1.0_gpt2_tmptoken_TRT_bs32_lr1e5_linearLR")