devin_lee commited on
Commit
89b03ce
·
1 Parent(s): d81a243

update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -25,9 +25,9 @@ def create_prompt_message(message: str) -> str:
25
  Message: {message}
26
  """.strip()
27
 
28
- def inference(model_name, message):
29
  # 根據 model_name 決定要用哪個模型
30
- model_id = model_list[model_name]
31
  model = openai(
32
  model_id,
33
  api_key=os.getenv("TOGETHER_API_KEY"),
@@ -51,7 +51,7 @@ def inference(model_name, message):
51
 
52
 
53
  with gr.Blocks() as demo:
54
- gr.Markdown("## LLM訂位意圖分類器🍴🥂")
55
  gr.Markdown("判斷輸入訊息是否與訂位有關")
56
 
57
  with gr.Row():
 
25
  Message: {message}
26
  """.strip()
27
 
28
+ def inference(model_dropdown, message):
29
  # 根據 model_name 決定要用哪個模型
30
+ model_id = model_list[model_dropdown]
31
  model = openai(
32
  model_id,
33
  api_key=os.getenv("TOGETHER_API_KEY"),
 
51
 
52
 
53
  with gr.Blocks() as demo:
54
+ gr.Markdown("## 🍴🥂 LLM訂位意圖分類器")
55
  gr.Markdown("判斷輸入訊息是否與訂位有關")
56
 
57
  with gr.Row():