Update app.py

app.py

@@ -3,11 +3,12 @@ import gradio as gr
 from transformers import T5ForConditionalGeneration, T5Tokenizer, pipeline
 
 # -----------------------------
-# 1. Load the Model & Slow Tokenizer
-#    We explicitly disable the fast tokenizer by setting use_fast=False.
+# 1. Load a Better Model & Slow Tokenizer
+#    Using the "valhalla/t5-base-qg-hl" model (a step up from the small version)
+#    and disabling the fast tokenizer.
 # -----------------------------
-tokenizer = T5Tokenizer.from_pretrained("valhalla/t5-small-qg-hl", use_fast=False)
-model = T5ForConditionalGeneration.from_pretrained("valhalla/t5-small-qg-hl")
+tokenizer = T5Tokenizer.from_pretrained("valhalla/t5-base-qg-hl", use_fast=False)
+model = T5ForConditionalGeneration.from_pretrained("valhalla/t5-base-qg-hl")
 qg_pipeline = pipeline(
     "text2text-generation",
     model=model,
@@ -108,16 +109,39 @@ def check_answer(state, user_answer):
     return result_text + "\n" + difficulty_update
 
 # -----------------------------
-# 7. Build the Gradio Interface
+# 7. Build the Gradio Interface with Custom CSS for a More Colorful UI
 # -----------------------------
-with gr.Blocks() as demo:
+custom_css = """
+body {
+    background: linear-gradient(135deg, #e0c3fc, #8ec5fc);
+    font-family: 'Helvetica', sans-serif;
+}
+.gradio-container {
+    background-color: #ffffff;
+    border-radius: 15px;
+    padding: 20px;
+    box-shadow: 0 4px 12px rgba(0,0,0,0.15);
+}
+.gradio-markdown h1 {
+    color: #4b0082;
+}
+.gradio-markdown p, .gradio-markdown li {
+    font-size: 16px;
+    color: #333333;
+}
+.gradio-button, .gradio-textbox {
+    font-size: 16px;
+}
+"""
+
+with gr.Blocks(css=custom_css) as demo:
     # Persistent state for the session.
     state = gr.State(init_state())
 
     gr.Markdown("# Adaptive Language Tutor")
     gr.Markdown(
         "This demo uses a T5-based model to generate questions from a passage. "
-        "Difficulty will automatically adjust based on your performance."
+        "The difficulty will automatically adjust based on your performance."
     )
 
     # Display current difficulty and score.