Update app.py

app.py

@@ -21,12 +21,8 @@ def load_model(model_name):
 # Initialize default model
 load_model("LeCarnet-8M")
 
-# Streamer for real-time generation
-streamer = None
-
 # Streaming generation function
 def respond(message, max_tokens, temperature, top_p):
-    global streamer
     inputs = tokenizer(message, return_tensors="pt")
     streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
 
@@ -68,10 +64,23 @@ def update_model(model_name):
 
 # Gradio UI
 with gr.Blocks(title="LeCarnet - Chat Interface") as demo:
+    # Top bar with logo and title
+    with gr.Row():
+        with gr.Column(min_width=100):
+            gr.Image(
+                value="media/le-carnet.png",
+                label="Model Logo",
+                height=60,
+                width=60,
+                interactive=False
+            )
+        with gr.Column():
+            gr.Markdown("# LeCarnet")
+
+    # Main layout
     with gr.Row():
         # Left column: Options
         with gr.Column(scale=1, min_width=150):
-            gr.Markdown("### 🧠 Model Settings")
             model_selector = gr.Dropdown(
                 choices=["LeCarnet-3M", "LeCarnet-8M", "LeCarnet-21M"],
                 value="LeCarnet-8M",
@@ -80,18 +89,10 @@ with gr.Blocks(title="LeCarnet - Chat Interface") as demo:
             max_tokens = gr.Slider(1, 512, value=512, step=1, label="Max New Tokens")
             temperature = gr.Slider(0.1, 2.0, value=0.7, step=0.1, label="Temperature")
             top_p = gr.Slider(0.1, 1.0, value=0.9, step=0.05, label="Top-p Sampling")
-            clear_button = gr.Button("🗑️ Clear Chat")
+            clear_button = gr.Button("Clear Chat")
 
-        # Right column: Chat + Image
+        # Right column: Chat
         with gr.Column(scale=4):
-            gr.Markdown("### 🤖 LeCarnet Chatbot")
-            model_logo = gr.Image(
-                value="media/le-carnet.png",
-                label="Model Logo",
-                height=100,
-                width=100,
-                interactive=False
-            )
             chatbot = gr.Chatbot(
                 bubble_full_width=False,
                 height=500