Update app.py

app.py

@@ -4,6 +4,7 @@ from stqdm import stqdm
 import pandas as pd
 from transformers import pipeline
 import json
+from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
 
 
@@ -123,20 +124,25 @@ def main():
     elif choice=="Question Generation":
         st.subheader("Question Generation")
         st.write(" Enter the text to get questions generated !")
-        question_generator = pipeline(model="mrm8488/t5-base-finetuned-question-generation-ap",tokenizer="mrm8488/t5-base-finetuned-question-generation-ap")
+        # Load the T5 model and tokenizer
+        model = AutoModelForSeq2SeqLM.from_pretrained("mrm8488/t5-base-finetuned-question-generation-ap")
+        tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-question-generation-ap")
         text_input2 = st.text_area("Your Text","Enter the Text to complete")
         
         
-        if text_input2:
-            # Extract named entities from the text
-            entities = extract_entities(text_input2)
-            # Generate questions based on the text using the T5 model
-            questions = question_generator(text_input2, max_length=30)
-            # Display the generated questions
-            st.subheader("Generated questions")
-            for question in questions:
-                st.write(question["question"])
+        # Create a button to generate questions
+        if st.button("Generate Questions"):
+            #Encode the input text using the tokenizer
+            input_ids = tokenizer.encode("generate questions: " + input_text, return_tensors="pt")
+            # Use the T5 model to generate questions
+            question_ids = model.generate(input_ids)
+            # Decode the questions from the output ids using the tokenizer
+            questions = tokenizer.decode(question_ids[0], skip_special_tokens=True)
+            # Display the questions to the user
+            st.write("Generated Questions:")
+            st.write(questions)
 
     #main function to run
 if __name__ == '__main__':
-	main() 
+	main() 
+