Update app.py

app.py

@@ -26,9 +26,9 @@ retriever = vectorstore.as_retriever()
 # 2. Load Model & Tokenizer in 4-bit
 # ==========================================
 print("Loading Model...")
-model_id = "anirudh248/llama3-upf-generator" # UPDATE THIS if your username is different
+model_id = "anirudh248/llama3-upf-generator"
 
-# Load in 4-bit to fit perfectly inside a T4 GPU Space
+# Load in 4-bit to fit inside a T4 GPU Space
 bnb_config = BitsAndBytesConfig(
     load_in_4bit=True,
     bnb_4bit_compute_dtype=torch.float16
@@ -92,10 +92,11 @@ def user_interaction(user_message, history):
     history.append({"role": "assistant", "content": response})
     return history, ""
 
-with gr.Blocks(theme=gr.themes.Soft()) as interface:
+with gr.Blocks() as interface:
     gr.Markdown("# ⚡ UPF Code Generator with Llama 3 & RAG")
     
-    chatbot = gr.Chatbot(label="Chat History", elem_id="chatbot", type="messages")
+    # Removed `type="messages"` parameter to fix TypeError
+    chatbot = gr.Chatbot(label="Chat History", elem_id="chatbot")
     
     with gr.Row():
         user_input = gr.Textbox(show_label=False, placeholder="Enter power intent...", lines=3)
@@ -105,4 +106,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as interface:
     send_button.click(fn=user_interaction, inputs=[user_input, chatbot], outputs=[chatbot, user_input])
     user_input.submit(fn=user_interaction, inputs=[user_input, chatbot], outputs=[chatbot, user_input])
 
-interface.launch()
+interface.launch(theme=gr.themes.Soft())