Sakalti commited on
Commit
dfed880
·
verified ·
1 Parent(s): 6736769

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
- from huggingface_hub import InferenceClient
3
  import os
 
4
 
5
  # 使用可能なモデルのリスト
6
  models = ["Sakalti/Saba1.5-Pro", "Sakalti/light-3b-beta", "Sakalti/Neptuno-Alpha",
@@ -13,7 +13,7 @@ system_message_templates = {
13
  "架空のキャラ アオイ": "あなたの名前はアオイ。常識的な考えを持っている。語尾は「だぜ」",
14
  }
15
 
16
- # Hugging Faceのアクセストークン(Hugging Faceの設定から取得してください)
17
  HF_TOKEN = os.getenv("HF_TOKEN")
18
 
19
  def update_system_message(selected_template):
@@ -30,6 +30,10 @@ def respond(
30
  ):
31
  selected_model = str(selected_model)
32
 
 
 
 
 
33
  # トークンを設定してInferenceClientを初期化
34
  client = InferenceClient(selected_model, token=HF_TOKEN)
35
 
@@ -87,7 +91,7 @@ with gr.Blocks(theme='Nymbo/Alyx_theme') as demo:
87
  additional_inputs=[
88
  system_message_textbox,
89
  gr.Slider(minimum=1, maximum=2048, value=768, step=1, label="新規トークン最大"),
90
- gr.Slider(minimum=0.1, maxgetenvimum=4.0, value=0.7, step=0.1, label="温度"),
91
  gr.Slider(
92
  minimum=0.1,
93
  maximum=1.0,
 
1
  import gradio as gr
 
2
  import os
3
+ from huggingface_hub import InferenceClient
4
 
5
  # 使用可能なモデルのリスト
6
  models = ["Sakalti/Saba1.5-Pro", "Sakalti/light-3b-beta", "Sakalti/Neptuno-Alpha",
 
13
  "架空のキャラ アオイ": "あなたの名前はアオイ。常識的な考えを持っている。語尾は「だぜ」",
14
  }
15
 
16
+ # 環境変数からHugging Faceのアクセストークンを取得
17
  HF_TOKEN = os.getenv("HF_TOKEN")
18
 
19
  def update_system_message(selected_template):
 
30
  ):
31
  selected_model = str(selected_model)
32
 
33
+ # トークンが取得できなかった場合のエラーハンドリング
34
+ if HF_TOKEN is None:
35
+ return "エラー: Hugging Faceのアクセストークンが設定されていません。環境変数`HF_TOKEN`を設定してください。"
36
+
37
  # トークンを設定してInferenceClientを初期化
38
  client = InferenceClient(selected_model, token=HF_TOKEN)
39
 
 
91
  additional_inputs=[
92
  system_message_textbox,
93
  gr.Slider(minimum=1, maximum=2048, value=768, step=1, label="新規トークン最大"),
94
+ gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="温度"),
95
  gr.Slider(
96
  minimum=0.1,
97
  maximum=1.0,