ashwml commited on
Commit
47ebffb
·
1 Parent(s): a9803d1
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -4,9 +4,9 @@ from transformers import Trainer, TrainingArguments
4
  from os.path import dirname
5
 
6
 
7
- model_output_path="model_output"
8
- my_model = GPT2LMHeadModel.from_pretrained(dirname)
9
- my_tokenizer = GPT2Tokenizer.from_pretrained(dirname)
10
 
11
  def predict(prompt):
12
  response = generate_response(my_model, my_tokenizer, prompt)
 
4
  from os.path import dirname
5
 
6
 
7
+ model_output_path="change"
8
+ my_model = GPT2LMHeadModel.from_pretrained(model_output_path)
9
+ my_tokenizer = GPT2Tokenizer.from_pretrained(model_output_path)
10
 
11
  def predict(prompt):
12
  response = generate_response(my_model, my_tokenizer, prompt)