Sakalti commited on
Commit
d594974
1 Parent(s): 4c23b86

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -7,7 +7,7 @@ from PIL import Image
7
  """
8
  Hugging Face Hubの推論APIについての詳細は、以下のドキュメントを参照してください: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
9
  """
10
- client = InferenceClient("Sakalti/SabaVL1-2B") # モデル名を更新
11
 
12
  def encode_image(image):
13
  buffered = BytesIO()
@@ -24,6 +24,9 @@ def respond(
24
  temperature,
25
  top_p,
26
  ):
 
 
 
27
  messages = [{"role": "system", "content": system_message}]
28
 
29
  for val in history:
@@ -56,7 +59,7 @@ def respond(
56
  gradioのChatInterfaceのカスタマイズについては、以下のドキュメントを参照してください: https://www.gradio.app/docs/chatinterface
57
  """
58
  demo = gr.ChatInterface(
59
- respond,
60
  additional_inputs=[
61
  gr.Image(type="pil", label="画像をアップロード"),
62
  gr.Textbox(value="あなたは親切なチャットボットです。", label="システムメッセージ"),
@@ -70,6 +73,9 @@ demo = gr.ChatInterface(
70
  label="Top-p (nucleus sampling)",
71
  ),
72
  ],
 
 
 
73
  )
74
 
75
  if __name__ == "__main__":
 
7
  """
8
  Hugging Face Hubの推論APIについての詳細は、以下のドキュメントを参照してください: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
9
  """
10
+ client = InferenceClient("Sakalti/SabaVL1-2B")
11
 
12
  def encode_image(image):
13
  buffered = BytesIO()
 
24
  temperature,
25
  top_p,
26
  ):
27
+ if history is None:
28
+ history = []
29
+
30
  messages = [{"role": "system", "content": system_message}]
31
 
32
  for val in history:
 
59
  gradioのChatInterfaceのカスタマイズについては、以下のドキュメントを参照してください: https://www.gradio.app/docs/chatinterface
60
  """
61
  demo = gr.ChatInterface(
62
+ fn=respond,
63
  additional_inputs=[
64
  gr.Image(type="pil", label="画像をアップロード"),
65
  gr.Textbox(value="あなたは親切なチャットボットです。", label="システムメッセージ"),
 
73
  label="Top-p (nucleus sampling)",
74
  ),
75
  ],
76
+ chatbot=gr.Chatbot(),
77
+ textbox=gr.Textbox(),
78
+ additional_inputs_before_textbox=True,
79
  )
80
 
81
  if __name__ == "__main__":