jolchmo commited on
Commit
dcc2576
·
1 Parent(s): 160e3f2
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import spaces
3
  import torch
4
  from transformers import AutoTokenizer, AutoModelForCausalLM
5
 
6
- model_name = "TheFinAI/Fin-o1-14B"
7
  tokenizer = AutoTokenizer.from_pretrained(model_name)
8
  model = AutoModelForCausalLM.from_pretrained(model_name)
9
 
@@ -18,7 +18,7 @@ def chat(input_text):
18
 
19
  demo = gr.Interface(
20
  fn=chat,
21
- inputs=gr.Textbox(label="输入问题", placeholder="例如: What is the ?"),
22
  outputs=gr.Textbox(label="回答"),
23
  title="FinChatBox",
24
  description="使用 TheFinAI 模型进行对话"
 
3
  import torch
4
  from transformers import AutoTokenizer, AutoModelForCausalLM
5
 
6
+ model_name = "TheFinAI/Fino1-8B"
7
  tokenizer = AutoTokenizer.from_pretrained(model_name)
8
  model = AutoModelForCausalLM.from_pretrained(model_name)
9
 
 
18
 
19
  demo = gr.Interface(
20
  fn=chat,
21
+ inputs=gr.Textbox(label="输入问题", placeholder="例如: What is the macd ?"),
22
  outputs=gr.Textbox(label="回答"),
23
  title="FinChatBox",
24
  description="使用 TheFinAI 模型进行对话"