Update app.py

app.py

@@ -5,10 +5,10 @@ from transformers import GemmaTokenizer, AutoModelForCausalLM
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 from threading import Thread
 
-
 # Set an environment variable
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
 
+
 DESCRIPTION = '''
 <div>
 <h1 style="text-align: center;">Meta Llama3 8B</h1>
@@ -25,21 +25,6 @@ LICENSE = """
 Built with Meta Llama 3
 """
 
-PLACEHOLDER1 = """
-<div style="opacity: 0.65;">
-    <img src="https://ysharma-dummy-chat-app.hf.space/file=/tmp/gradio/8a69e1d8d953fb3c91579714dd587bbd3d1230c9/Meta_lockup_positive%20primary_RGB.png" style="width:45%;">
-    <br><b>Meta Llama3-8B Chatbot</b>
-</div>
-"""
-
-PLACEHOLDER2 = """
-<div style="padding: 30px; text-align: center; display: flex; flex-direction: column; align-items: center;">
-   <img src="https://ysharma-dummy-chat-app.hf.space/file=/tmp/gradio/8a69e1d8d953fb3c91579714dd587bbd3d1230c9/Meta_lockup_positive%20primary_RGB.png" style="width: 80%; max-width: 450px; height: auto; opacity: 0.55; margin-bottom: 10px; border-radius: 10px; box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);"> 
-   <h1 style="font-size: 28px; margin-bottom: 2px; color: #000; opacity: 0.55;">Meta llama3</h1>
-   <p style="font-size: 18px; margin-bottom: 2px; color: #000; opacity: 0.65;">Ask me anything...</p>
-</div>
-"""
-
 PLACEHOLDER = """
 <div style="padding: 30px; text-align: center; display: flex; flex-direction: column; align-items: center;">
    <img src="https://ysharma-dummy-chat-app.hf.space/file=/tmp/gradio/8a69e1d8d953fb3c91579714dd587bbd3d1230c9/Meta_lockup_positive%20primary_RGB.png" style="width: 80%; max-width: 550px; height: auto; opacity: 0.55;  "> 
@@ -89,7 +74,6 @@ def chat_llama3_8b(message: str,
     conversation.append({"role": "user", "content": message})
 
     input_ids = tokenizer.apply_chat_template(conversation, return_tensors="pt").to(model.device)
-    #input_ids = tokenizer.encode(message, return_tensors="pt").to(model.device)
     
     streamer = TextIteratorStreamer(tokenizer, timeout=10.0, skip_prompt=True, skip_special_tokens=True)
 
@@ -115,7 +99,7 @@ def chat_llama3_8b(message: str,
         
 
 # Gradio block
-chatbot=gr.Chatbot(height=500, placeholder=PLACEHOLDER)
+chatbot=gr.Chatbot(height=500, placeholder=PLACEHOLDER, label='Gradio ChatInterface')
 
 with gr.Blocks(fill_height=True, css=css) as demo: