Ngadou commited on
Commit
d6ae027
·
1 Parent(s): f1a09d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -10,6 +10,8 @@ model = AutoModelForCausalLM.from_pretrained("vilsonrodrigues/falcon-7b-instruct
10
  tokenizer = AutoTokenizer.from_pretrained("vilsonrodrigues/falcon-7b-instruct-sharded")
11
  model = PeftModel.from_pretrained(model, "Ngadou/falcon-7b-scam-buster").to("cuda")
12
 
 
 
13
  def generate(chat):
14
 
15
  input_text = chat + "\nIs this conversation a scam or not and why?"
 
10
  tokenizer = AutoTokenizer.from_pretrained("vilsonrodrigues/falcon-7b-instruct-sharded")
11
  model = PeftModel.from_pretrained(model, "Ngadou/falcon-7b-scam-buster").to("cuda")
12
 
13
+ tokenizer.pad_token = tokenizer.eos_token
14
+
15
  def generate(chat):
16
 
17
  input_text = chat + "\nIs this conversation a scam or not and why?"