Spaces:
Runtime error
Runtime error
reshinthadithyan
commited on
Commit
·
a83a01c
1
Parent(s):
226c1ac
adding prompt style
Browse files
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,
|