chore: minor updates

app.py

@@ -38,7 +38,7 @@ model = Llama.from_pretrained(
     **model_paths[DEFAULT_MODEL],
     n_ctx=4096,
     n_threads=4,
-    cache_dir='./hf-cache'
+    cache_dir='./.hf_cache'
 )
 
 def generate_alpaca_prompt(
@@ -105,13 +105,13 @@ def on_model_changed(model_name: str):
         cache_dir='./hf-cache'
     )
 
-    app_title_mark = gr.Markdown(f"""<center><font size=16>{model_name}</center>""")
+    app_title_mark = gr.Markdown(f"""<center><font size=18>{model_name}</center>""")
     chatbot = gr.Chatbot(
         type='messages',
         height=500,
-        placeholder='<strong>Hi, I have a headache, what should I do?</strong>',
+        placeholder='<strong>Hi doctor, I have a headache, what should I do?</strong>',
         label=model_name,
-        avatar_images=[None, './assets/assistant_avavar.png'],  # pyright: ignore[reportArgumentType]
+        avatar_images=[None, 'https://raw.githubusercontent.com/minhnguyent546/medical-llama2/demo/assets/med_alpaca.png'],  # pyright: ignore[reportArgumentType]
     )
     return app_title_mark, chatbot
 
@@ -130,7 +130,7 @@ def main() -> None:
                     max_new_tokens = gr.Number(value=512, minimum=64, maximum=2048, label='Max new tokens')
 
                 with gr.Row():
-                    temperature = gr.Slider(0, 2, step=0.01, label='Temperature', value=0.6, info='Info')
+                    temperature = gr.Slider(0, 2, step=0.01, label='Temperature', value=0.6)
                     repeatition_penalty = gr.Slider(0.01, 5, step=0.05, label='Repetition penalty', value=1.1)
 
                 with gr.Row():
@@ -141,12 +141,12 @@ def main() -> None:
                 chatbot = gr.Chatbot(
                     type='messages',
                     height=500,
-                    placeholder='<strong>Hi, I have a headache, what should I do?</strong>',
+                    placeholder='<strong>Hi doctor, I have a headache, what should I do?</strong>',
                     label=DEFAULT_MODEL,
-                    avatar_images=[None, './assets/assistant_avavar.png'],  # pyright: ignore[reportArgumentType]
+                    avatar_images=[None, 'https://raw.githubusercontent.com/minhnguyent546/medical-llama2/demo/assets/med_alpaca.png'],  # pyright: ignore[reportArgumentType]
                 )
                 textbox = gr.Textbox(
-                    placeholder='Hi, I have a headache, what should I do?',
+                    placeholder='Hi doctor, I have a headache, what should I do?',
                     container=False,
                     submit_btn=True,
                     stop_btn=True,

requirements.txt

@@ -1,3 +1,4 @@
 gradio~=5.6.0
-huggingface_hub==0.25.2
+hf_transfer~=0.1.8
+huggingface_hub~=0.25.2
 llama-cpp-python~=0.3.2