Add local logo

hackaprompt/gradio_app.py

@@ -104,7 +104,12 @@ def get_current_prompt(level, current_user_prompts):
     return current_user_prompts[level]
 
 
-with gr.Blocks() as demo:
+with gr.Blocks(css="""
+    #logo {
+        max-width: 5%;
+        height: auto;
+    }
+""") as demo:
     # state to store user's prompts
     current_user_prompts = gr.State(["" for _ in range(NUM_LEVELS)])
 
@@ -117,12 +122,11 @@ with gr.Blocks() as demo:
     # Default evaluator when page loads (level 0)
     evaluator_0 = gr.State(get_evaluator(level=0, completer=None))
 
+    gr.Image(value="mila_logo.png", container=False, elem_id="logo", show_label=False, show_download_button=False, show_fullscreen_button=False)  # Adjust the size of the logo as needed
+
     gr.Markdown(
     """
-    <div style="display: flex; align-items: center;">
-         <img src="https://mila.quebec/sites/default/files/media-library/image/4032/milalogowebcoulrgb.png" width="120" style="margin-right: 0px;">
-         <h1 style="margin: 0;">Hands-on Exercise: Exploring Jailbreaking & Gen AI Risks</h1>
-    </div>
+    # Hands-on Exercise: Exploring Jailbreaking & Gen AI Risks
 
     This exercise is part of the TRAIL Responsible Gen AI Risk Module, designed to provide hands-on experience with the challenges of Gen AI. 
 
@@ -201,4 +205,4 @@ USERNAME, PASSWORD = os.getenv("APP_USERNAME"), os.getenv("APP_PASSWORD")
 if USERNAME and PASSWORD:
     demo.launch(auth=(USERNAME, PASSWORD))
 else:
-    demo.launch()
+    demo.launch(allowed_paths=["/hackaprompt/"])