supermy commited on
Commit
4be04dc
·
1 Parent(s): d93236c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def top_k_top_p_filtering( logits, top_k=0, top_p=0.0, filter_value=-float('Inf'
27
 
28
  def generate(title, context, max_len):
29
 
30
- input_ids.extend( tokenizer.encode(title + "-" + context, add_special_tokens=False) )
31
 
32
  # title_ids = tokenizer.encode(title, add_special_tokens=False)
33
  # context_ids = tokenizer.encode(context, add_special_tokens=False)
 
27
 
28
  def generate(title, context, max_len):
29
 
30
+ input_ids=tokenizer.encode(title + "-" + context, add_special_tokens=False)
31
 
32
  # title_ids = tokenizer.encode(title, add_special_tokens=False)
33
  # context_ids = tokenizer.encode(context, add_special_tokens=False)