File size: 1,064 Bytes
46bef1e
e648b17
46bef1e
917e741
e648b17
6baf09a
b40d855
 
e648b17
 
 
 
 
b40d855
e648b17
 
 
b40d855
e648b17
 
 
 
b40d855
e648b17
 
 
b40d855
e648b17
 
 
 
b40d855
46bef1e
 
4ac79c2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
46bef1e
e648b17
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
import gradio as gr
import os
import time
import google.generativeai as palm

palm.configure(api_key=os.environ.get("palm_key"))

defaults = {
    'model': 'models/chat-bison-001',
    'temperature': 0.25,
    'candidate_count': 1,
    'top_k': 40,
    'top_p': 0.95,
}

context = "You're a computer failure assistant"

examples = [
    [
        "Hey my computer is broken",
        "Hey, what is the issue with your computer?"
    ]
]

messages = ["NEXT REQUEST"]

response = palm.chat(
    **defaults,
    context=context,
    examples=examples,
    messages=messages
)





with gr.Blocks() as demo:
    chatbot = gr.Chatbot()
    msg = gr.Textbox()
    clear = gr.ClearButton([msg, chatbot])

    def respond(msg):
        messages.append(msg)
        bot_message = palm.chat(
            **defaults,
            context=context,
            examples=examples,
            messages=messages
        )
        time.sleep(2)
        return bot_message['message'], messages

    msg.submit(respond, [msg, chatbot], [msg, chatbot])

demo.launch()

iface.launch()