ysharma HF staff commited on
Commit
f40416e
1 Parent(s): e477eb0

added avatar support in chat

Browse files
Files changed (1) hide show
  1. app_dialogue.py +3 -0
app_dialogue.py CHANGED
@@ -71,6 +71,8 @@ PROCESSOR = AutoProcessor.from_pretrained(
71
  token=API_TOKEN,
72
  )
73
 
 
 
74
  logging.basicConfig(level=logging.INFO)
75
  logger = logging.getLogger()
76
 
@@ -375,6 +377,7 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
375
  label="IDEFICS",
376
  visible=True,
377
  height=750,
 
378
  # value=[
379
  # [
380
  # (
 
71
  token=API_TOKEN,
72
  )
73
 
74
+ BOT_AVATAR = "IDEFICS_logo.png"
75
+
76
  logging.basicConfig(level=logging.INFO)
77
  logger = logging.getLogger()
78
 
 
377
  label="IDEFICS",
378
  visible=True,
379
  height=750,
380
+ avatar_images=[None, BOT_AVATAR]
381
  # value=[
382
  # [
383
  # (