Spaces:
Sleeping
Sleeping
File size: 2,669 Bytes
735784f 13df059 a986897 fee4a07 36694ba 6b17378 36694ba ca7fd27 36694ba 56205d5 919188b ca7fd27 56205d5 ca7fd27 9d8af5f ca7fd27 8a82df0 658baf1 82e5470 56205d5 148cfd6 800b433 148cfd6 9d8af5f ba21a6a |
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 |
import subprocess
subprocess.check_call(["pip", "install", "openai"])
subprocess.check_call(["pip", "install", "gradio", "transformers", "python-dotenv","torch"])
import torch
import random
import gradio as gr
from transformers import TFAutoModelForCausalLM, AutoTokenizer
import openai
from dotenv import load_dotenv
import os
load_dotenv() # load environment variables from .env file
api_key = os.getenv("OPENAI_API_KEY") # access the value of the OPENAI_API_KEY environment variable
'''
def openai_chat(prompt):
if "who are you" in prompt.lower() or "your name" in prompt.lower() or "name" in prompt.lower():
return "My name is ChatSherman. How can I assist you today?"
else:
prompt = "I'm an AI chatbot named ChatSherman designed by a super intelligent student named ShermanAI at the Department of Electronic and Information Engineering at the Hong Kong Polytechnic University to help you with your engineering questions. Also, I can assist with a wide range of topics and questions." + prompt
completions = openai.Completion.create(engine="text-davinci-003", prompt=prompt, max_tokens=10000, n=1, temperature=0.5,)
message = completions.choices[0].text
return message.strip()
'''
def getresponse(message, history):
system_prompt = "I'm an AI chatbot named ChatSherman designed by a super intelligent student named ShermanAI at the Department of Electronic and Information Engineering at the Hong Kong Polytechnic University to help you with your engineering questions. Also, I can assist with a wide range of topics and questions."
messages = [{"role":"system","content":system_prompt}]
for human, assistant in history:
messages.append({"role":"user", "content":human})
messages.append({"role":"assistant", "content":assistant})
if message != '':
messages.append({"role":"user", "content":message})
response = openai.ChatCompletion.create(engine = "text-davinci-003",
messages = messages,
temperature =0.5,
max_tokens = 10000,
top_p = 0.95,
frequency_penalty = 1,
presence_penalty = 1,
stop = None)
return response["choices"][0]["message"]["content"]
title = "ChatSherman"
description = "This is an AI chatbot powered by ShermanAI. Enter your question below to get started."
examples = [
["What is ChatSherman, and how does it work?"],
["Is my personal information and data safe when I use the ChatSherman chatbot?"],
["What are some common applications of deep learning in engineering?"]
]
gr.Interface(getresponse, title=title, description=description, examples=examples).launch() # ChatInterface |