ifire commited on
Commit
b7f1c62
1 Parent(s): c37d4b6

Restore model access.

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,8 +15,8 @@ import gradio as gr
15
 
16
  from transformers import AutoTokenizer, AutoConfig, AutoModelForCausalLM
17
 
18
- finetuned = AutoModelForCausalLM.from_pretrained('model')
19
- tokenizer = AutoTokenizer.from_pretrained('gpt2')
20
 
21
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
22
  print(device)
 
15
 
16
  from transformers import AutoTokenizer, AutoConfig, AutoModelForCausalLM
17
 
18
+ tokenizer = AutoTokenizer.from_pretrained("architext/gptj-162M")
19
+ finetuned = AutoModelForCausalLM.from_pretrained("architext/gptj-162M")
20
 
21
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
22
  print(device)