Lyiosvf commited on
Commit
8d50969
·
verified ·
1 Parent(s): 4f356d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -12
app.py CHANGED
@@ -1,26 +1,27 @@
1
  import gradio as gr
2
- #import random
3
  from huggingface_hub import InferenceClient
4
 
5
  client = InferenceClient("microsoft/phi-4")
6
 
7
  def respond(message, history):
8
-
9
- messages = [{"role":"system", "content": "You are a friendly chatbot!"}]
10
-
11
  if history:
12
  messages.extend(history)
13
- messages.append({"role": "user", "content":message})
14
-
15
- response = client.chat_completion(messages, max_tokens=100)
16
-
17
 
18
- #responses = ["Yes","No"] #added here
19
- return response['choices'][0]['message']['content'].strip() #added here
20
-
21
- chatbot = gr.ChatInterface(respond, type="messages", title="Friendly chatbot", ) #chatbot UI - conversion history and user input
 
 
22
 
 
23
 
 
24
 
25
  custom_css = """
26
  body {
 
1
  import gradio as gr
 
2
  from huggingface_hub import InferenceClient
3
 
4
  client = InferenceClient("microsoft/phi-4")
5
 
6
  def respond(message, history):
7
+
8
+ messages = [{"role": "system", "content": "You are a friendly chatbot."}]
9
+
10
  if history:
11
  messages.extend(history)
12
+
13
+ messages.append({"role": "user", "content": message})
 
 
14
 
15
+ response = client.chat_completion(
16
+ messages,
17
+ max_tokens=100
18
+ )
19
+
20
+ return response['choices'][0]['message']['content'].strip()
21
 
22
+ chatbot = gr.ChatInterface(respond, type="messages", title="KpopBot")
23
 
24
+ chatbot.launch()
25
 
26
  custom_css = """
27
  body {