File size: 1,809 Bytes
78e2d06
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5639ffc
 
 
78e2d06
 
 
 
3545fc9
78e2d06
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
import openai
import gradio as gr

openai.api_key = "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"  

def predict(message, history):
    history_openai_format = []
    for human, assistant in history:
        history_openai_format.append({"role": "user", "content": human })
        history_openai_format.append({"role": "assistant", "content":assistant})
    history_openai_format.append({"role": "user", "content": message})

    response = openai.ChatCompletion.create(
        model='gpt-3.5-turbo',
        messages= history_openai_format,
        temperature=1.0,
        stream=True
    )

    partial_message = ""
    for chunk in response:
        if len(chunk['choices'][0]['delta']) != 0:
            partial_message = partial_message + chunk['choices'][0]['delta']['content']
            yield partial_message

A1 = gr.ChatInterface(predict,
                title="PeachTalk+",
                description="An AI Powered Chatbot with Computer Vision and Image Generation Capabilities Currently Under Development By Peach State Innovation and Technology. Ask Me About Question About Anything...From Georgia and Beyond...And I'll Give You An Answer!",
                theme= gr.themes.Glass(primary_hue="amber", neutral_hue="lime"),
                retry_btn=None,
                clear_btn="Clear")

A2 = gr.load(
             "huggingface/Salesforce/blip-image-captioning-large",
              title="Upon Further Review...",
              description="Upload or Take a Photo Image, I'll Give You Description",
              theme= gr.themes.Glass(primary_hue="amber", neutral_hue="lime"))

A3 = gr.load("huggingface/google/vit-base-patch16-224")

pcp = gr.TabbedInterface([A1, A2], ["Chat", "Describe"], theme= gr.themes.Glass(primary_hue="amber", neutral_hue="lime"))
pcp.queue().launch()