Update app.py

app.py

@@ -16,6 +16,7 @@ DEFAULT_MAX_NEW_TOKENS = 1024
 MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "8192"))
 
 model_id = "BramVanroy/fietje-2b-chat"
+avatar_url = "https://huggingface.co/spaces/BramVanroy/fietje-2b/resolve/main/img/fietje-2b-avatar.png"
 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto")
 model = model.to("cuda" if torch.cuda.is_available() else "cpu")
 tokenizer = AutoTokenizer.from_pretrained(model_id)
@@ -93,7 +94,7 @@ chat_interface = gr.ChatInterface(
     chatbot=gr.Chatbot(height=450,
                       label="Fietje",
                       show_share_button=True,
-                      avatar_images=(None, 'fietje-2b-avatar.png')),
+                      avatar_images=(None, avatar_url)),
     additional_inputs=[
         gr.Slider(
             label="Max new tokens",
@@ -146,9 +147,9 @@ chat_interface = gr.ChatInterface(
         ["Wat is het belang van open-source taalmodellen?"],
     ],
     title="Fietje",
-    description="""\
+    description=f"""\
 <div style="float: left; margin-right: 1em; margin-bottom: 0.48em; max-width: 96px;">
-    <img src="fietje-2b-banner.png" style="display: inline-block; width: auto; height: auto;border-radius:50%">
+    <img src={avatar_url} style="display: inline-block; width: auto; height: auto;border-radius:50%">
     <div style="clear: both"></div>
 </div>