Update app.py

app.py

@@ -27,9 +27,15 @@ def top_k_top_p_filtering( logits, top_k=0, top_p=0.0, filter_value=-float('Inf'
     return logits
 
 def generate(title, context, max_len):
+
+    # input_ids.extend( tokenizer.encode(input_text + "-", add_special_tokens=False) )
+
     title_ids = tokenizer.encode(title, add_special_tokens=False)
     context_ids = tokenizer.encode(context, add_special_tokens=False)
+    print(title_ids,context_ids)
+    
     input_ids = title_ids + [sep_id] + context_ids
+    
     cur_len = len(input_ids)
     input_len = cur_len
     last_token_id = input_ids[-1]
@@ -60,7 +66,6 @@ if __name__ == '__main__':
     eod_id = tokenizer.convert_tokens_to_ids("<eod>")
     sep_id = tokenizer.sep_token_id
     unk_id = tokenizer.unk_token_id
-
  
     
     gr.Interface(
@@ -71,4 +76,4 @@ if __name__ == '__main__':
             "number"
             ],
         outputs=gr.Textbox(lines=15, placeholder="AI生成的文本显示在这里。",label="生成的文本")
-    ).launch()
+    ).launch()