Update app.py

app.py

@@ -1,13 +1,13 @@
 import streamlit as st
 from transformers import pipeline
 
-# Load the Hugging Face model for question generation
-question_generator = pipeline("text2text-generation", model="valhalla/t5-small-qg-hl")
+# Load the Hugging Face model for text generation
+pipe = pipeline("text-generation", model="google/gemma-2-27b-it")
 
 # Function to generate MCQs
 def generate_mcqs(content, num_questions):
-    # Generate questions with beam search
-    questions = question_generator(content, max_length=512, num_return_sequences=num_questions, num_beams=num_questions)
+    messages = [{"role": "user", "content": content}]
+    questions = pipe(messages, max_length=512, num_return_sequences=num_questions, num_beams=num_questions)
     return questions
 
 # Streamlit UI
@@ -27,3 +27,4 @@ if st.button("Generate MCQs"):
 
 
 
+