sammanamgain commited on
Commit
555a102
1 Parent(s): ea5da90

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -11,9 +11,8 @@ model.config.pad_token_id = model.config.eos_token_id
11
  def generate_response(prompt):
12
  # Use a special separator token or pattern
13
  instruction = "how to open a account"
14
- category = "ACCOUNT"
15
- intent = ""
16
- input_text = f"{prompt} {category} objective:{intent} [SEP]"
17
 
18
  # Tokenize the input and create the attention mask
19
  inputs = tokenizer(input_text, return_tensors="pt", padding=True, truncation=True)
 
11
  def generate_response(prompt):
12
  # Use a special separator token or pattern
13
  instruction = "how to open a account"
14
+
15
+ input_text = f"{prompt}"
 
16
 
17
  # Tokenize the input and create the attention mask
18
  inputs = tokenizer(input_text, return_tensors="pt", padding=True, truncation=True)