ashwml commited on
Commit
b2aec5f
·
1 Parent(s): 243b015

edited app file

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import Trainer, TrainingArguments
4
 
5
 
6
 
7
- model_output_path='/output'
8
  my_model = GPT2LMHeadModel.from_pretrained(model_output_path)
9
  my_tokenizer = GPT2Tokenizer.from_pretrained(model_output_path)
10
 
 
4
 
5
 
6
 
7
+ model_output_path="/repo"
8
  my_model = GPT2LMHeadModel.from_pretrained(model_output_path)
9
  my_tokenizer = GPT2Tokenizer.from_pretrained(model_output_path)
10