File size: 761 Bytes
c231631
b0ee8dd
f6b2521
 
 
 
 
 
c231631
b0ee8dd
8b3849b
 
 
 
0588558
f6b2521
89b2356
f6b2521
 
 
 
 
 
70d8d54
f6b2521
 
c231631
8b3849b
f6b2521
c231631
 
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
import gradio as gr
import random 
import os

from huggingface_hub import InferenceClient


client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")

#Chat responds yes or no randomly
#def respond_yes_no_randomly(message,history):
   # responses =["Yes", "No"]
   # return random.choice(responses)

#Chatbot is a magic 8 ball
def respond(message,history):
    messages = [{"role" : "system" , "content" : "College advisor."}]
    if history:
        messages.extend(history)

    messages.append({"role" : "user", "content" : message}) 
    response = client.chat_completion(
        messages,
        max_tokens=300
    )
    return response['choices'][0]['message']['content'].strip()


chatbot = gr.ChatInterface(respond, type = "messages")

chatbot.launch()