yakul259 commited on
Commit
3c5170e
·
verified ·
1 Parent(s): ebf8891

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +34 -23
app.py CHANGED
@@ -11,61 +11,72 @@ def respond(
11
  top_p,
12
  hf_token: gr.OAuthToken,
13
  ):
14
- """
15
- For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
16
- """
17
- client = InferenceClient(token=hf_token.token, model="openai/gpt-oss-20b")
 
18
 
19
- messages = [{"role": "system", "content": system_message}]
 
 
 
 
 
 
20
 
 
 
 
21
  messages.extend(history)
22
 
 
23
  messages.append({"role": "user", "content": message})
24
 
25
  response = ""
26
 
27
- for message in client.chat_completion(
 
28
  messages,
29
  max_tokens=max_tokens,
30
  stream=True,
31
  temperature=temperature,
32
  top_p=top_p,
33
  ):
34
- choices = message.choices
35
  token = ""
36
- if len(choices) and choices[0].delta.content:
37
- token = choices[0].delta.content
38
 
39
  response += token
40
  yield response
41
 
42
 
43
- """
44
- For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
45
- """
 
46
  chatbot = gr.ChatInterface(
47
  respond,
48
  additional_inputs=[
49
- gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
50
- gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
51
- gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
52
- gr.Slider(
53
- minimum=0.1,
54
- maximum=1.0,
55
- value=0.95,
56
- step=0.05,
57
- label="Top-p (nucleus sampling)",
58
  ),
 
 
 
59
  ],
60
  )
61
 
62
  with gr.Blocks() as demo:
 
 
 
63
  with gr.Sidebar():
64
  gr.LoginButton()
 
65
  chatbot.render()
66
 
67
 
68
  if __name__ == "__main__":
69
  demo.launch()
70
-
71
- this is the app.py already given as a template in gradio so give your code accordingly
 
11
  top_p,
12
  hf_token: gr.OAuthToken,
13
  ):
14
+ # Initialize client (model kept internal)
15
+ client = InferenceClient(
16
+ token=hf_token.token,
17
+ model="openai/gpt-oss-20b"
18
+ )
19
 
20
+ # 🔥 Strong hidden finance instruction
21
+ system_prompt = (
22
+ "You are a professional financial advisor. "
23
+ "Answer all questions related to finance, investments, risk, valuation, and markets. "
24
+ "Provide clear explanations, use proper financial terminology, "
25
+ "and include step-by-step calculations when required."
26
+ )
27
 
28
+ messages = [{"role": "system", "content": system_prompt}]
29
+
30
+ # Add chat history
31
  messages.extend(history)
32
 
33
+ # Add user message
34
  messages.append({"role": "user", "content": message})
35
 
36
  response = ""
37
 
38
+ # Streaming response
39
+ for msg in client.chat_completion(
40
  messages,
41
  max_tokens=max_tokens,
42
  stream=True,
43
  temperature=temperature,
44
  top_p=top_p,
45
  ):
 
46
  token = ""
47
+ if msg.choices and msg.choices[0].delta.content:
48
+ token = msg.choices[0].delta.content
49
 
50
  response += token
51
  yield response
52
 
53
 
54
+ # ==============================
55
+ # UI
56
+ # ==============================
57
+
58
  chatbot = gr.ChatInterface(
59
  respond,
60
  additional_inputs=[
61
+ gr.Textbox(
62
+ value="Ask any finance-related question.",
63
+ label="Instructions"
 
 
 
 
 
 
64
  ),
65
+ gr.Slider(1, 2048, value=512, step=1, label="Response Length"),
66
+ gr.Slider(0.1, 2.0, value=0.7, step=0.1, label="Creativity"),
67
+ gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="Response Diversity"),
68
  ],
69
  )
70
 
71
  with gr.Blocks() as demo:
72
+ gr.Markdown("# 💰 Finance Chatbot")
73
+ gr.Markdown("Ask questions about investments, risk, valuation, and financial concepts.")
74
+
75
  with gr.Sidebar():
76
  gr.LoginButton()
77
+
78
  chatbot.render()
79
 
80
 
81
  if __name__ == "__main__":
82
  demo.launch()