Hugging Face's logo

Spaces:
Ayemos
/
highlight_text_based_on_eye_movement
Runtime error

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

add model

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -6,8 +6,15 @@ import torch
6
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
7
 
8
  device = torch.device("cuda:0") if torch.cuda.is_available() else torch.device("cpu")
 
9
  tokenizer = GPT2Tokenizer.from_pretrained("dendee-geco_test-on-zuco1.0_gpt2_tmptoken_TRT_bs32_lr1e6_linearLR")
10
  model = GPT2LMHeadModel.from_pretrained("dendee-geco_test-on-zuco1.0_gpt2_tmptoken_TRT_bs32_lr1e6_linearLR", return_dict=True)
 
 
 
 
 
 
11
  model.to(device)
12
 
13
 
 
6
  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")
15
+ model = GPT2LMHeadModel.from_pretrained("zuco1.0_gpt2_tmptoken_TRT_bs32_lr1e5_linearLR", return_dict=True)
16
+
17
+
18
  model.to(device)
19
 
20