Update app.py

app.py

@@ -1,35 +1,6 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-# Load the tokenizer and model
-model_name = "mohamedemam/QA_GeneraToR"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-
-# Recommended words for users to choose from
-recommended_words = [
-    "which", "how", "when", "where", "who", "whom", "whose", "why",
-    "which", "who", "whom", "whose", "whereas",
-    "can", "could", "may", "might", "will", "would", "shall", "should",
-    "do", "does", "did", "is", "are", "am", "was", "were", "be", "being", "been",
-    "have", "has", "had", "if", "is", "are", "am", "was", "were", "do", "does", "did", "can", "could",
-    "will", "would", "shall", "should", "might", "may", "must",
-    "may", "might", "must",
-]
-
-#
-example_contexts=[  "when: Lionel Andrés Messi[note 1] (Spanish pronunciation: [ljoˈnel anˈdɾes ˈmesi] (listen); born 24 June 1987), also known as Leo Messi, is an Argentine professional footballer who plays as a forward for and captains both Major League Soccer club Inter Miami and the Argentina national team. Widely regarded as one of the greatest players of all time, Messi has won a record seven Ballon d'Or awards[note 2] and a record six European Golden Shoes, and in 2020 he was named to the Ballon d'Or Dream Team. Until leaving the club in 2021, he had spent his entire professional career with Barcelona, where he won a club-record 34"
- , "where: Lionel Andrés Messi[note 1] (Spanish pronunciation: [ljoˈnel anˈdɾes ˈmesi] (listen); born 24 June 1987), also known as Leo Messi, is an Argentine professional footballer who plays as a forward for and captains both Major League Soccer club Inter Miami and the Argentina national team. Widely regarded as one of the greatest players of all time, Messi has won a record seven Ballon d'Or awards[note 2] and a record six European Golden Shoes, and in 2020 he was named to the Ballon d'Or Dream Team. Until leaving the club in 2021, he had spent his entire professional career with Barcelona, where he won a club-record 34"
- , "how: Lionel Andrés Messi[note 1] (Spanish pronunciation: [ljoˈnel anˈdɾes ˈmesi] (listen); born 24 June 1987), also known as Leo Messi, is an Argentine professional footballer who plays as a forward for and captains both Major League Soccer club Inter Miami and the Argentina national team. Widely regarded as one of the greatest players of all time, Messi has won a record seven Ballon d'Or awards[note 2] and a record six European Golden Shoes, and in 2020 he was named to the Ballon d'Or Dream Team. Until leaving the club in 2021, he had spent his entire professional career with Barcelona, where he won a club-record 34"
- , "what: Lionel Andrés Messi[note 1] (Spanish pronunciation: [ljoˈnel anˈdɾes ˈmesi] (listen); born 24 June 1987), also known as Leo Messi, is an Argentine professional footballer who plays as a forward for and captains both Major League Soccer club Inter Miami and the Argentina national team. Widely regarded as one of the greatest players of all time, Messi has won a record seven Ballon d'Or awards[note 2] and a record six European Golden Shoes, and in 2020 he was named to the Ballon d'Or Dream Team. Until leaving the club in 2021, he had spent his entire professional career with Barcelona, where he won a club-record 34"
- ,"where: Egypt (Egyptian Arabic: مصر Maṣr Egyptian Arabic pronunciation: [mɑsˤr]), officially the Arab Republic of Egypt, is a transcontinental country spanning the northeast corner of Africa and the Sinai Peninsula in the southwest corner of Asia. It is bordered by the Mediterranean Sea to the north, the Gaza Strip of Palestine and Israel to the northeast, the Red Sea to the east, Sudan to the south, and Libya to the west. The Gulf of Aqaba in the northeast separates Egypt from Jordan and Saudi Arabia. Cairo is the capital and largest city of Egypt, while Alexandria, the second-largest city, is an important industrial and tourist hub at the Mediterranean coast.[11] At approximately 100 million inhabitants, Egypt is the 14th-most populated country in the world, and the third-most populated in Africa, behind Nigeria and Ethiopia."
-, "where: There is evidence of rock carvings along the Nile terraces and in desert oases. In the 10th millennium BCE, a culture of hunter-gatherers and fishers was replaced by a grain-grinding culture. Climate changes or overgrazing around 8000 BCE began to desiccate the pastoral lands of Egypt, forming the Sahara. Early tribal peoples migrated to the Nile River where they developed a settled agricultural economy and more centralized society."
- , "when: By about 6000 BCE, a Neolithic culture took root in the Nile Valley.[31] During the Neolithic era, several predynastic cultures developed independently in Upper and Lower Egypt. The Badarian culture and the successor Naqada series are generally regarded as precursors to dynastic Egypt. The earliest known Lower Egyptian site, Merimda, predates the Badarian by about seven hundred years. Contemporaneous Lower Egyptian communities coexisted with their southern counterparts for more than two thousand years. The earliest known evidence of Egyptian hieroglyphic inscriptions appeared during the predynastic period on Naqada III pottery vessels, dated to about 3200 BCE.[32]"
-, "whose : or the next three millennia. Egyptian culture flourished during this long period and remained distinctively Egyptian in its religion, arts, language and customs. The first two ruling dynasties of a unified Egypt set the stage for the Old Kingdom period, c. 2700–2200 BCE, which constructed many pyramids, most notably the Third Dynasty pyramid of Djoser and the Fourth Dynasty Giza pyramids."
-, "who:The First Intermediate Period ushered in a time of political upheaval for about 150 years.[33] Stronger Nile floods and stabilisation of government, however, brought back renewed prosperity for the country in the Middle Kingdom c. 2040 BCE, reaching a peak during the reign of Pharaoh Amenemhat III. A second period of disunity heralded the arrival of the first foreign ruling dynasty in Egypt, that of the Semitic Hyksos. The Hyksos invaders took over much of Lower Egypt around 1650 BCE and founded a new capital at Avaris. They were driven out by an Upper Egyptian force led by Ahmose I, who founded the Eighteenth Dynasty and relocated the capital from Memphis to Thebes."]
-import gradio as gr
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-
 # Load the tokenizer and model
 model_name = "mohamedemam/QA_GeneraTor"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -43,6 +14,17 @@ recommended_words = [
     "have", "has", "had", "if", "must",
 ]
 
+# Example contexts
+example_contexts = [
+    "when: Lionel Andrés Messi...",
+    "where: Lionel Andrés Messi...",
+    "how: Lionel Andrés Messi...",
+    "what: Lionel Andrés Messi...",
+    "where: Egypt...",
+    "where: There is evidence..."
+    # Add more examples here
+]
+
 # Function to generate questions and answers with configurable parameters
 def generate_qa(context, recommended_word, temperature, top_p, num_samples=3):
     input_text = f"{recommended_word}: {context}"
@@ -53,26 +35,29 @@ def generate_qa(context, recommended_word, temperature, top_p, num_samples=3):
         **input_ids,
         temperature=temperature,
         top_p=top_p,
-num_return_sequences=num_samples,do_sample=True,    top_k=0
-
+        num_return_sequences=3,
+        do_sample=True,
+        max_length=100,
+        num_beams=6,
+        length_penalty=1,
+        top_k=0
     )
     
     generated_text = tokenizer.batch_decode(output, skip_special_tokens=True)
     return generated_text
 
-# Create the Gradio interface with sliders for temperature and top-p
 # Create the Gradio interface with sliders for temperature and top-p
 iface = gr.Interface(
     fn=generate_qa,
     inputs=[
-        "text",
+        gr.inputs.Dropdown(example_contexts, label="Choose an Example"),
         gr.inputs.Radio(recommended_words, label="Choose a Recommended Word"),
-        gr.inputs.Slider(minimum=0.2, maximum=2, default=1, step=0.1, label="Temperature"),
-        gr.inputs.Slider(minimum=0.1, maximum=1, default=0.8, step=0.1, label="Top-p")
+        gr.inputs.Slider(minimum=0.0, maximum=2, default=2.1, step=0.1, label="Temperature"),
+        gr.inputs.Slider(minimum=0.0, maximum=1, default=0.5, step=0.1, label="Top-p")
     ],
-    outputs="text",
+    outputs=gr.outputs.Table(columns=["Generated Sentences"]),
     title="Question Generation and Answering",
-    description="Enter a context, choose a recommended word, and adjust temperature and top-p. The model will generate a question and answer.",
+    description="Select an example context, choose a recommended word, adjust temperature and top-p. The model will generate questions and answers.",
 )
 
 # Launch the interface