s3nh commited on
Commit
3f33715
1 Parent(s): 4679d7b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -46,7 +46,7 @@ def generate_prompt(instruction, input=None):
46
  generation_config = create_generator()
47
 
48
 
49
- def evaluate(instruction, input=None):
50
  prompt = generate_prompt(instruction, input)
51
  inputs = tokenizer(prompt, return_tensors="pt")
52
  input_ids = inputs["input_ids"]
@@ -66,7 +66,7 @@ def evaluate(instruction, input=None):
66
  def inference(model_name, text, input):
67
  model = load_model(model_name)
68
  tokenizer = load_tokenizer(model_name)
69
- output = evaluate(instruction = text, input = input)
70
  return output
71
 
72
  def choose_model(name):
 
46
  generation_config = create_generator()
47
 
48
 
49
+ def evaluate(instruction, input=None, model, tokenizer):
50
  prompt = generate_prompt(instruction, input)
51
  inputs = tokenizer(prompt, return_tensors="pt")
52
  input_ids = inputs["input_ids"]
 
66
  def inference(model_name, text, input):
67
  model = load_model(model_name)
68
  tokenizer = load_tokenizer(model_name)
69
+ output = evaluate(instruction = text, input = input, model = model, tokenizer = tokenizer)
70
  return output
71
 
72
  def choose_model(name):