Update UI

app.py

@@ -9,7 +9,7 @@ HF_TOKEN = os.environ.get("HF_TOKEN", None)
 
 DESCRIPTION = '''
 <div>
-<h1 style="text-align: center;">StevenChen16 Llama3 8B Lawyer</h1>
+<h1 style="text-align: center;">AI Lawyer</h1>
 <p>This Space demonstrates the instruction-tuned model <a href="https://huggingface.co/StevenChen16/llama3-8b-Lawyer"><b>StevenChen16 Llama3 8B Lawyer</b></a>. This model is fine-tuned to provide legal advice based on US and Canada law.</p>
 <p>Feel free to play with it, or duplicate to run privately!</p>
 </div>
@@ -18,12 +18,12 @@ DESCRIPTION = '''
 LICENSE = """
 <p/>
 ---
-Built with StevenChen16 Llama3 8B Lawyer
+Built with model "StevenChen16/Llama3-8B-Lawyer", based on "meta-llama/Meta-Llama-3-8B"
 """
 
 PLACEHOLDER = """
 <div style="padding: 30px; text-align: center; display: flex; flex-direction: column; align-items: center;">
-   <h1 style="font-size: 28px; margin-bottom: 2px; opacity: 0.55;">StevenChen16 Llama3 8B Lawyer</h1>
+   <h1 style="font-size: 28px; margin-bottom: 2px; opacity: 0.55;">AI Lawyer</h1>
    <p style="font-size: 18px; margin-bottom: 2px; opacity: 0.65;">Ask me anything about US and Canada law...</p>
 </div>
 """
@@ -60,12 +60,12 @@ You are an advanced AI legal assistant trained to assist with a wide range of le
 5. **Professional Tone**: Maintain a professional and respectful tone in all responses. Ensure that your advice is legally sound and adheres to ethical standards.
 """
 
-def query_model(user_input, history, temperature, max_new_tokens):
+def query_model(user_input, history):
     combined_query = background_prompt + user_input
     messages = [{"role": "user", "content": combined_query}]
     
     response = ""
-    for new_text in chat_model.stream_chat(messages):
+    for new_text in chat_model.stream_chat(messages, max_new_tokens=512, temperature=0.9):
         response += new_text
         yield response
     
@@ -78,20 +78,16 @@ with gr.Blocks(css=css) as demo:
     gr.ChatInterface(
         fn=query_model,
         chatbot=chatbot,
-        additional_inputs=[
-            gr.Slider(minimum=0, maximum=1, step=0.1, value=0.95, label="Temperature"),
-            gr.Slider(minimum=128, maximum=4096, step=1, value=512, label="Max new tokens"),
-        ],
         examples=[
-            ['How to setup a human base on Mars? Give short answer.'],
-            ['Explain theory of relativity to me like I’m 8 years old.'],
-            ['What is 9,000 * 9,000?'],
-            ['Write a pun-filled happy birthday message to my friend Alex.'],
-            ['Justify why a penguin might make a good king of the jungle.']
+            ['What are the key differences between a sole proprietorship and a partnership?'],
+            ['What legal steps should I take if I want to start a business in the US?'],
+            ['Can you explain the concept of "duty of care" in negligence law?'],
+            ['What are the legal requirements for obtaining a patent in Canada?'],
+            ['How can I protect my intellectual property when sharing my idea with potential investors?']
         ],
         cache_examples=False,
     )
     gr.Markdown(LICENSE)
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch()