Update app.py

app.py

@@ -9,6 +9,8 @@ import time # Import time module for potential debugging/delay
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 genai.configure(api_key=GEMINI_API_KEY)
 
+used_model = "gemini-2.0-flash-thinking-exp-01-21"
+
 # we will be using the Gemini 2.0 Flash model with Thinking capabilities
 model = genai.GenerativeModel("gemini-2.0-flash-thinking-exp-01-21")
 
@@ -134,7 +136,7 @@ def user_message(msg: str, history: list) -> tuple[str, list]:
 
 # Create the Gradio interface
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal", secondary_hue="slate", neutral_hue="neutral")) as demo: # Using Soft theme with adjusted hues for a refined look
-    gr.Markdown("# Chat with Gemini 2.0 Flash and See its Thoughts 💭")
+    gr.Markdown("# Chat with " + used_model)
 
     
     gr.HTML("""<a href="https://visitorbadge.io/status?path=https%3A%2F%2Faiqcamp-Gemini2-Flash-Thinking.hf.space">