CasperDylan commited on
Commit
e86aaf4
·
verified ·
1 Parent(s): 951960f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +27 -1
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import os
2
  import subprocess
 
3
  import gradio as gr
4
 
5
  # 安裝 groq 套件
@@ -24,4 +25,29 @@ def chatbot_response(user_message):
24
  ]
25
 
26
  completion = client.chat.completions.create(
27
- model
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import os
2
  import subprocess
3
+ import sys
4
  import gradio as gr
5
 
6
  # 安裝 groq 套件
 
25
  ]
26
 
27
  completion = client.chat.completions.create(
28
+ model="llama-3.1-70b-versatile",
29
+ messages=messages,
30
+ temperature=1,
31
+ max_tokens=1024,
32
+ top_p=1,
33
+ stream=False,
34
+ stop=None
35
+ )
36
+
37
+ response_message = completion.choices[0].message['content']
38
+ return response_message
39
+
40
+ # 建立 Gradio 界面
41
+ with gr.Blocks() as demo:
42
+ chatbot = gr.Chatbot()
43
+ user_input = gr.Textbox(placeholder="輸入訊息...")
44
+ send_button = gr.Button("發送")
45
+
46
+ def user_message_handler(user_message, history):
47
+ bot_response = chatbot_response(user_message)
48
+ history.append((user_message, bot_response))
49
+ return history, ""
50
+
51
+ send_button.click(user_message_handler, [user_input, chatbot], [chatbot, user_input])
52
+
53
+ demo.launch()