BharathNarasimha commited on
Commit
54ecf3a
1 Parent(s): 35e97ec

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +33 -0
app.py ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import gradio as gr
3
+ from langchain.chat_models import ChatOpenAI
4
+ from langchain import LLMChain, PromptTemplate
5
+ from langchain.memory import ConversationBufferMemory
6
+
7
+ OPENAI_API_KEY=os.getenv('OPENAI_API_KEY')
8
+
9
+ template = """You are a sports-loving high school student with a keen interest in multiple sports, from soccer and basketball to tennis and swimming. You closely follow sports events, stats, and news, making you the go-to person for all sports-related discussions and predictions.
10
+ User: {user_message}
11
+ Chatbot:"""
12
+
13
+ prompt = PromptTemplate(
14
+ input_variables=["chat_history", "user_message"], template=template
15
+ )
16
+
17
+ memory = ConversationBufferMemory(memory_key="chat_history")
18
+
19
+ llm_chain = LLMChain(
20
+ llm=ChatOpenAI(temperature='0.5', model_name="gpt-3.5-turbo"),
21
+ prompt=prompt,
22
+ verbose=True,
23
+ memory=memory,
24
+ )
25
+
26
+ def get_text_response(user_message,history):
27
+ response = llm_chain.predict(user_message = user_message)
28
+ return response
29
+
30
+ demo = gr.ChatInterface(get_text_response)
31
+
32
+ if __name__ == "__main__":
33
+ demo.launch() #To create a public link, set `share=True` in `launch()`. To enable errors and logs, set `debug=True` in `launch()`.