devin_lee commited on
Commit
7483ca3
·
1 Parent(s): 333a04d

update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -19,7 +19,7 @@ def load_classifier(model_key):
19
  )
20
  return choice(model, ["Reservation", "Others"])
21
 
22
- classifier = load_classifier("Gemma 2B")
23
 
24
  def create_prompt_message(message: str) -> str:
25
  return f"""
@@ -71,7 +71,7 @@ with gr.Blocks() as demo:
71
  outputs=gr.Textbox(label="分類結果")
72
  with gr.Row():
73
  inference_btn = gr.Button("開始分類")
74
-
75
  inference_btn.click(
76
  fn=inference,
77
  inputs=inputs,
@@ -79,4 +79,3 @@ with gr.Blocks() as demo:
79
  )
80
 
81
  demo.launch()
82
-
 
19
  )
20
  return choice(model, ["Reservation", "Others"])
21
 
22
+ classifier = load_classifier("Gemma-2B")
23
 
24
  def create_prompt_message(message: str) -> str:
25
  return f"""
 
71
  outputs=gr.Textbox(label="分類結果")
72
  with gr.Row():
73
  inference_btn = gr.Button("開始分類")
74
+
75
  inference_btn.click(
76
  fn=inference,
77
  inputs=inputs,
 
79
  )
80
 
81
  demo.launch()