Leyo commited on
Commit
02dcc48
1 Parent(s): 7cae97c

Update app_dialogue.py

Browse files
Files changed (1) hide show
  1. app_dialogue.py +36 -1
app_dialogue.py CHANGED
@@ -56,6 +56,41 @@ SYSTEM_PROMPT = [
56
  ],
57
  }
58
  ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
59
 
60
  API_TOKEN = os.getenv("HF_AUTH_TOKEN")
61
  HF_WRITE_TOKEN = os.getenv("HF_WRITE_TOKEN")
@@ -454,7 +489,7 @@ with gr.Blocks(
454
  gr.ChatInterface(
455
  fn=model_inference,
456
  chatbot=chatbot,
457
- # examples=[{"text": "hello"}, {"text": "hola"}, {"text": "merhaba"}],
458
  title="Idefics2 Playground",
459
  multimodal=True,
460
  additional_inputs=[
 
56
  ],
57
  }
58
  ]
59
+ examples_path = os.path.dirname(__file__)
60
+ EXAMPLES = [
61
+ [
62
+ {
63
+ "text": "Describe the image",
64
+ "files": [f"{examples_path}/example_images/plant_bulb.webp"],
65
+ }
66
+ ],
67
+ [
68
+ {
69
+ "text": "Describe the image",
70
+ "files": [f"{examples_path}/example_images/pope_doudoune.webp"],
71
+ }
72
+ ],
73
+ [
74
+ {
75
+ "text": "Describe the image",
76
+ "files": [
77
+ f"{examples_path}/example_images/kittens-cats-pet-cute-preview.jpg"
78
+ ],
79
+ }
80
+ ],
81
+ [
82
+ {
83
+ "text": "Describe the image",
84
+ "files": [f"{examples_path}/example_images/baguettes_guarding_paris.png"],
85
+ }
86
+ ],
87
+ [
88
+ {
89
+ "text": "Describe the image",
90
+ "files": [f"{examples_path}/example_images/dragons_playing.png"],
91
+ }
92
+ ],
93
+ ]
94
 
95
  API_TOKEN = os.getenv("HF_AUTH_TOKEN")
96
  HF_WRITE_TOKEN = os.getenv("HF_WRITE_TOKEN")
 
489
  gr.ChatInterface(
490
  fn=model_inference,
491
  chatbot=chatbot,
492
+ examples=EXAMPLES,
493
  title="Idefics2 Playground",
494
  multimodal=True,
495
  additional_inputs=[