Update app.py

app.py

@@ -1,7 +1,6 @@
 import os
 import gradio as gr
 import torch
-import tiktoken
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 from threading import Thread
 
@@ -25,10 +24,6 @@ model = AutoModelForCausalLM.from_pretrained(
 print("Model loaded successfully.")
 
 def generate_code(prompt, history):
-    # Gradio 6.x uses the 'messages' format by default
-    # history will look like: [{"role": "user", "content": "..."}, {"role": "assistant", "content": "..."}]
-    
-    # Prepend existing history to the current prompt
     messages = history + [{"role": "user", "content": prompt}]
     
     # Prepare inputs using the model's chat template
@@ -63,7 +58,6 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as demo:
     
     # In Gradio 6, type="messages" is the standard for the chatbot component
     chatbot = gr.Chatbot(height=500, show_copy_button=True, type="messages")
-    
     with gr.Row():
         msg = gr.Textbox(
             placeholder="E.g., Create a responsive navigation bar with CSS...", 
@@ -71,12 +65,8 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as demo:
             scale=9
         )
         submit = gr.Button("Build", variant="primary", scale=1)
-
-    # Wire up the events
     msg.submit(generate_code, [msg, chatbot], [chatbot])
     submit.click(generate_code, [msg, chatbot], [chatbot])
-    
-    # Clear the input box after submission
     msg.submit(lambda: "", None, [msg])
     submit.click(lambda: "", None, [msg])