Update app.py

app.py

@@ -18,6 +18,14 @@ This Space demonstrates model [CataLlama-v0.1-Instruct-DPO](https://huggingface.
 CataLlama is a fine-tune on Llama-3 8B to enhance it's proficiency on the Catalan Language.
 """
 
+LICENSE = """
+<p/>
+---
+As a derivate work of [Meta-Llama-3-8B](https://huggingface.co/meta-llama/Meta-Llama-3-8B) by Meta,
+this demo is governed by the original [llama-3 license](https://llama.meta.com/llama3/license)
+
+"""
+
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
 
@@ -33,9 +41,9 @@ def generate(
     message: str,
     chat_history: list[tuple[str, str]],
     system_prompt: str,
-    max_new_tokens: int = 1024,
-    temperature: float = 0.6,
-    top_p: float = 0.9,
+    max_new_tokens: int,
+    temperature: float,
+    top_p: float,
 ) -> Iterator[str]:
     
     conversation = []
@@ -75,9 +83,25 @@ def generate(
 chat_interface = gr.ChatInterface(
     fn=generate,
     additional_inputs=[
-        gr.Textbox(value="Ets un chatbot amigable. Responeu preguntes i ajudeu els usuaris", label="System message"),
-        gr.Slider(minimum=1, maximum=2048, value=1024, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.6, step=0.1, label="Temperature"),
+        gr.Textbox(
+            value="Ets un chatbot amigable. Responeu preguntes i ajudeu els usuaris", 
+            label="System message", 
+            lines=6
+        ),
+        gr.Slider(
+            minimum=1, 
+            maximum=2048, 
+            value=1024, 
+            step=256, 
+            label="Max new tokens"
+        ),
+        gr.Slider(
+            minimum=0.1, 
+            maximum=1.0, 
+            value=0.6, 
+            step=0.05, 
+            label="Temperature"
+        ),
         gr.Slider(
             minimum=0.1,
             maximum=1.0,
@@ -96,6 +120,7 @@ chat_interface = gr.ChatInterface(
 with gr.Blocks(css="style.css") as demo:
     gr.Markdown(DESCRIPTION)
     chat_interface.render()
+    gr.Markdown(LICENSE)
 
 
 if __name__ == "__main__":