yangswei commited on
Commit
1040dfe
·
1 Parent(s): b9ea338

Add application file

Browse files
Files changed (1) hide show
  1. song-insight-app.py +3 -1
song-insight-app.py CHANGED
@@ -6,6 +6,8 @@ from langchain_community.retrievers import WikipediaRetriever
6
  import os
7
 
8
 
 
 
9
  def song_meaning(song, artist):
10
  artist_input = artist.title()
11
  song_input = song.title()
@@ -26,7 +28,7 @@ def song_meaning(song, artist):
26
  prompt_template_song_meaning = PromptTemplate(input_variables=["artist", "song", "content"],
27
  template=template_song_meaning)
28
 
29
- llm = ChatOpenAI(openai_api_key=os.getenv('OPENAI_API_KEY'), model_name="gpt-3.5-turbo", temperature=0)
30
  chain = LLMChain(llm=llm, prompt=prompt_template_song_meaning)
31
  results = chain.run(artist=artist_input, song=song_input, content=docs[0].page_content)
32
 
 
6
  import os
7
 
8
 
9
+ OPENAI_API_KEY = os.environ['OPENAI_API_KEY']
10
+
11
  def song_meaning(song, artist):
12
  artist_input = artist.title()
13
  song_input = song.title()
 
28
  prompt_template_song_meaning = PromptTemplate(input_variables=["artist", "song", "content"],
29
  template=template_song_meaning)
30
 
31
+ llm = ChatOpenAI(openai_api_key=OPENAI_API_KEY, model_name="gpt-3.5-turbo", temperature=0)
32
  chain = LLMChain(llm=llm, prompt=prompt_template_song_meaning)
33
  results = chain.run(artist=artist_input, song=song_input, content=docs[0].page_content)
34