Update app.py

app.py

@@ -62,33 +62,6 @@ if 'generated' not in st.session_state:
 if 'past' not in st.session_state:
     st.session_state['past'] = []
 
-def query(history):
-    inputs = tokenizer.dialogue_encode(
-        history, add_start_token_as_response=True, return_tensors=True, is_split_into_words=False
-    )
-    inputs["input_ids"] = inputs["input_ids"].astype("int64")
-    ids, scores = model.generate(
-        input_ids=inputs["input_ids"],
-        token_type_ids=inputs["token_type_ids"],
-        position_ids=inputs["position_ids"],
-        attention_mask=inputs["attention_mask"],
-        max_length=64,
-        min_length=1,
-        decode_strategy="sampling",
-        temperature=1.0,
-        top_k=5,
-        top_p=1.0,
-        num_beams=0,
-        length_penalty=1.0,
-        early_stopping=False,
-        num_return_sequences=20,
-    )
-    max_dec_len = 64
-    num_return_sequences = 20
-    bot_response = select_response(
-        ids, scores, tokenizer, max_dec_len, num_return_sequences, keep_space=False
-    )[0]
-    return bot_response
 
 def get_text():
     input_text = st.text_input("用户: ","你好！", key="input")