Sakalti commited on
Commit
5ff012b
·
verified ·
1 Parent(s): d594974

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -15
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()
@@ -58,24 +58,25 @@ def respond(
58
  """
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="システムメッセージ"),
66
- gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
67
- gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
68
- gr.Slider(
69
- minimum=0.1,
70
- maximum=1.0,
71
- value=0.95,
72
- step=0.05,
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__":
 
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") # モデル名をQwen2-VL-2B-Instructに更新
11
 
12
  def encode_image(image):
13
  buffered = BytesIO()
 
58
  """
59
  gradioのChatInterfaceのカスタマイズについては、以下のドキュメントを参照してください: https://www.gradio.app/docs/chatinterface
60
  """
61
+ additional_inputs = [
62
+ gr.Image(type="pil", label="画像をアップロード"),
63
+ gr.Textbox(value="あなたは親切なチャットボットです。", label="システムメッセージ"),
64
+ gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
65
+ gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
66
+ gr.Slider(
67
+ minimum=0.1,
68
+ maximum=1.0,
69
+ value=0.95,
70
+ step=0.05,
71
+ label="Top-p (nucleus sampling)",
72
+ ),
73
+ ]
74
+
75
  demo = gr.ChatInterface(
76
  fn=respond,
77
+ additional_inputs=additional_inputs,
 
 
 
 
 
 
 
 
 
 
 
 
78
  chatbot=gr.Chatbot(),
79
  textbox=gr.Textbox(),
 
80
  )
81
 
82
  if __name__ == "__main__":