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

Add application file

Browse files
Files changed (1) hide show
  1. song-insight-app.py +1 -3
song-insight-app.py CHANGED
@@ -6,8 +6,6 @@ from langchain_community.retrievers import WikipediaRetriever
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,7 +26,7 @@ def song_meaning(song, artist):
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
 
 
6
  import os
7
 
8
 
 
 
9
  def song_meaning(song, artist):
10
  artist_input = artist.title()
11
  song_input = song.title()
 
26
  prompt_template_song_meaning = PromptTemplate(input_variables=["artist", "song", "content"],
27
  template=template_song_meaning)
28
 
29
+ llm = ChatOpenAI(openai_api_key="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", 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