Rob Learsch commited on
Commit
fe45956
·
1 Parent(s): 7281021

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -103,8 +103,8 @@ def artist_response(gemma_response, artist):
103
  artist_embeddings = grateful_dead_embeddings
104
  lyric_list = all_phrases_grateful_dead
105
  #lyric_list = stitched_grateful_dead_lyrics
106
- #if artist == "Google Gemma":
107
- # return gemma_response
108
 
109
  encoded_gemma = encoder_model.encode(gemma_response)
110
  similarity_result = encoder_model.similarity(
@@ -141,8 +141,8 @@ def chat_with_musician(user_input, history, artist):
141
  history.append({"role": "assistant", "content": gemma_response})
142
 
143
  lyric_response = artist_response(gemma_response, artist)
144
- if artist == "Google Gemma":
145
- lyric_response = gemma_response
146
  return lyric_response
147
 
148
  radiohead_lyrics = load_lyrics("radiohead_lyrics.txt")
 
103
  artist_embeddings = grateful_dead_embeddings
104
  lyric_list = all_phrases_grateful_dead
105
  #lyric_list = stitched_grateful_dead_lyrics
106
+ if artist == "Google Gemma":
107
+ return gemma_response
108
 
109
  encoded_gemma = encoder_model.encode(gemma_response)
110
  similarity_result = encoder_model.similarity(
 
141
  history.append({"role": "assistant", "content": gemma_response})
142
 
143
  lyric_response = artist_response(gemma_response, artist)
144
+ #if artist == "Google Gemma":
145
+ # lyric_response = gemma_response
146
  return lyric_response
147
 
148
  radiohead_lyrics = load_lyrics("radiohead_lyrics.txt")