reshinthadithyan commited on
Commit
a83a01c
·
1 Parent(s): 226c1ac

adding prompt style

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -14,6 +14,7 @@ tokenizer,model = load_model_and_tokenizer(MODEL_NAME)
14
  MAX_TOKS = 128
15
  MAX_NEW_TOKS = 128
16
  def generate_text(prompt):
 
17
  inputs = tokenizer(prompt, truncation=True, return_tensors="pt")
18
  output_seq = model.generate(
19
  input_ids=inputs.input_ids, max_length=MAX_TOKS,
 
14
  MAX_TOKS = 128
15
  MAX_NEW_TOKS = 128
16
  def generate_text(prompt):
17
+ prompt = "<english> " + prompt + " <bash>"
18
  inputs = tokenizer(prompt, truncation=True, return_tensors="pt")
19
  output_seq = model.generate(
20
  input_ids=inputs.input_ids, max_length=MAX_TOKS,