test

app.py

@@ -82,8 +82,7 @@ with gr.Blocks() as demo:
     #     time.sleep(2)
     #     return "", chat_history
     
-    def respond_entities(message, chat_history):
-        entity = entity_dropdown.value
+    def respond_entities(entity, message, chat_history):
         prompt = template_single.format(entity, message)
         input_ids = tokenizer.encode(prompt, return_tensors="pt")
         output_ids = model.generate(input_ids, max_length=50, num_beams=5, no_repeat_ngram_size=2)