rahul7star commited on
Commit
cb7c3ec
·
verified ·
1 Parent(s): 644cfe8

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +57 -0
app.py ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from huggingface_hub import InferenceClient
3
+
4
+ client = InferenceClient("rahul7star/rahul7star-ai-bot-model-v1")
5
+
6
+ title = "rahul7star chat bot. "
7
+
8
+ def respond(
9
+ message,
10
+ history: list[tuple[str, str]],
11
+ system_message,
12
+ max_tokens,
13
+ temperature,
14
+ top_p,
15
+ ):
16
+ messages = [{"role": "system", "content": system_message}]
17
+
18
+ for val in history:
19
+ if val[0]:
20
+ messages.append({"role": "user", "content": val[0]})
21
+ if val[1]:
22
+ messages.append({"role": "assistant", "content": val[1]})
23
+
24
+ messages.append({"role": "user", "content": message + title})
25
+
26
+ response = title + ""
27
+
28
+ for message in client.chat_completion(
29
+ messages,
30
+ max_tokens=max_tokens,
31
+ stream=True,
32
+ temperature=temperature,
33
+ top_p=top_p,
34
+ ):
35
+ token = message.choices[0].delta.content
36
+
37
+ response += token
38
+ yield response
39
+
40
+ demo = gr.ChatInterface(
41
+ respond,
42
+ additional_inputs=[
43
+ gr.Textbox(value="I am rahul Chatbot.", label="System message"),
44
+ gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
45
+ gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
46
+ gr.Slider(
47
+ minimum=0.1,
48
+ maximum=1.0,
49
+ value=0.95,
50
+ step=0.05,
51
+ label="Top-p (nucleus sampling)",
52
+ ),
53
+ ],
54
+ )
55
+
56
+ if __name__ == "__main__":
57
+ demo.launch()