Update app.py

app.py

@@ -56,7 +56,7 @@ recommended_words = [
 ]
 
 # Function to generate questions and answers with configurable parameters
-def generate_qa(context, recommended_word, temperature, top_p,num_seq, num_samples=3):
+def generate_qa(context, recommended_word, temperature, top_p,num_seq,l_p, num_b):
     input_text = f"{recommended_word}: {context}"
     input_text=re.sub(f'\n'," ",input_text).lower()
     input_ids = tokenizer(input_text, return_tensors='pt')
@@ -67,10 +67,10 @@ def generate_qa(context, recommended_word, temperature, top_p,num_seq, num_sampl
         temperature=temperature,
         top_p=top_p,
         num_return_sequences=num_seq,
-        do_sample=True,
+   
         max_length=100,
-        num_beams=6,
-        length_penalty=1.5,
+        num_beams=num_b,
+        length_penalty=l_p,     do_sample=True,
     )
     #
     generated_text = tokenizer.batch_decode(output, skip_special_tokens=True)
@@ -84,7 +84,12 @@ iface = gr.Interface(
         gr.inputs.Radio(recommended_words, label="Choose a Recommended Word"),
         gr.inputs.Slider(minimum=0.0, maximum=5, default=2.1, step=0.01, label="Temperature"),
         gr.inputs.Slider(minimum=0.0, maximum=1, default=0.5, step=0.01, label="Top-p"),     
-        gr.inputs.Slider(minimum=1, maximum=20, default=3, step=1, label="num of sequance")
+        gr.inputs.Slider(minimum=1, maximum=20, default=3, step=1, label="num of sequance"), 
+        gr.inputs.Slider(minimum=1, maximum=20, default=3, step=1, label="l_p")
+     ,
+        gr.inputs.Slider(minimum=1, maximum=20, default=3, step=1, label="num of beams"),
+
+
     ],
     outputs=gr.outputs.Textbox(label="Generated Output"),
     title="Question Generation and Answering",