Update app.py

app.py

@@ -205,6 +205,14 @@ def generate_quiz(question_difficulty, topic):
         return ['Please enter a valid topic.'] + [gr.Radio(visible=False) for _ in range(10)]
   
     top_k_rank = 10
+    # Load the model and database within the generate_quiz function
+    try:
+        RAG = RAGPretrainedModel.from_pretrained("colbert-ir/colbertv2.0")
+        RAG_db_ = RAG.from_index('.ragatouille/colbert/indexes/cbseclass10index')
+        gr.Warning('Model loaded!')
+    except Exception as e:
+        return [f"Error loading model: {e}"] + [gr.Radio(visible=False) for _ in range(10)]
+  
     RAG_db_ = RAG_db.value
     documents_full = RAG_db_.search(topic, k=top_k_rank)
     
@@ -378,9 +386,16 @@ with gr.Blocks(title="Quiz Maker", theme=gr.themes.Default(primary_hue="green",
         """)
     
     with gr.Column(scale=2):
-        load_btn = gr.Button("Click to Load!🚀")
-        load_text = gr.Textbox()
-        load_btn.click(fn=load_model, outputs=load_text)
+        gr.HTML("""
+        <center>
+            
+            <h2>Ready!</h2>
+            
+        </center>
+        """)
+        # load_btn = gr.Button("Click to Load!🚀")
+        # load_text = gr.Textbox()
+        # load_btn.click(fn=load_model, outputs=load_text)
     
     topic = gr.Textbox(label="Enter the Topic for Quiz", placeholder="Write any topic/details from Customs Manual")