Update app.py
Browse files
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):
|