update app.py

app.py

@@ -648,7 +648,8 @@ def averitec_search(claim, generate_question, speaker="they", check_date="2024-0
     store_folder = "averitec/data/store/retrieved_docs"
     #
     index = 0
-    questions = [q["question"] for q in generate_question]
+    questions = [q["question"] for q in generate_question][:5]
+    # questions = [q["question"] for q in generate_question]    # ori
 
     # check the date of the claim
     current_date = datetime.now().strftime("%Y-%m-%d")
@@ -783,8 +784,8 @@ def decorate_with_questions(claim, retrieve_evidence, top_k=3):  # top_k=5, 10,
 
         inputs = qg_tokenizer(sentences, padding=True, return_tensors="pt").to(qg_model.device)
         # inputs = qg_tokenizer(sentences, padding=True, return_tensors="pt").to(device)
-        outputs = qg_model.generate(inputs["input_ids"], max_length=2000, num_beams=2, no_repeat_ngram_size=2, early_stopping=True)
-                                                        # max_length=5000
+        outputs = qg_model.generate(inputs["input_ids"], max_length=5000, num_beams=2, no_repeat_ngram_size=2, early_stopping=True)
+
         tgt_text = qg_tokenizer.batch_decode(outputs[:, inputs["input_ids"].shape[-1]:], skip_special_tokens=True)[0]
         # We are not allowed to generate more than 250 characters:
         tgt_text = tgt_text[:250]
@@ -1423,8 +1424,8 @@ def main():
         dropdown_samples.change(change_sample_questions, dropdown_samples, samples)
         demo.queue()
 
-    # demo.launch()
-    demo.launch(share=True)
+    demo.launch()
+    # demo.launch(share=True)
 
 
 if __name__ == "__main__":