Crimson206 commited on
Commit
31ab7aa
Β·
verified Β·
1 Parent(s): 7ef3e29

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +20 -12
app.py CHANGED
@@ -1,18 +1,26 @@
 
1
  import gradio as gr
2
- import time
3
 
4
- def echo(message, history, system_prompt, tokens):
5
- response = f"System prompt: {system_prompt}\n Message: {message}."
6
- for i in range(min(len(response), int(tokens))):
7
- time.sleep(0.05)
8
- yield response[: i+1]
9
 
10
- with gr.Blocks() as demo:
11
- system_prompt = gr.Textbox("You are helpful AI.", label="System Prompt")
12
- slider = gr.Slider(10, 100, render=False)
 
 
 
13
 
14
- gr.ChatInterface(
15
- echo, additional_inputs=[system_prompt, slider]
 
 
 
16
  )
17
 
18
- demo.launch()
 
 
 
 
 
 
 
1
+ import openai
2
  import gradio as gr
 
3
 
4
+ openai.api_key = "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" # Replace with your key
 
 
 
 
5
 
6
+ def predict(message, history):
7
+ history_openai_format = []
8
+ for human, assistant in history:
9
+ history_openai_format.append({"role": "user", "content": human })
10
+ history_openai_format.append({"role": "assistant", "content":assistant})
11
+ history_openai_format.append({"role": "user", "content": message})
12
 
13
+ response = openai.ChatCompletion.create(
14
+ model='gpt-3.5-turbo',
15
+ messages= history_openai_format,
16
+ temperature=1.0,
17
+ stream=True
18
  )
19
 
20
+ partial_message = ""
21
+ for chunk in response:
22
+ if len(chunk['choices'][0]['delta']) != 0:
23
+ partial_message = partial_message + chunk['choices'][0]['delta']['content']
24
+ yield partial_message
25
+
26
+ gr.ChatInterface(predict).launch()