Rob Learsch commited on
Commit
a8c9f05
·
1 Parent(s): 5e55500

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -130,7 +130,7 @@ def chat_with_musician(user_input, history, artist):
130
  messages=messages,
131
  model="google/gemma-2-2b-it",
132
  max_tokens=256,
133
- temperature=1,
134
  #top_p=0.9
135
  )
136
  gemma_response= response["choices"][0]["message"]["content"]
@@ -170,7 +170,8 @@ encoder_model = SentenceTransformer('all-MiniLM-L6-v2',
170
  radiohead_embeddings = encoder_model.encode(all_phrases_radiohead,
171
  #precision="binary",
172
  show_progress_bar=True,)
173
- #kendrick_embeddings = encoder_model.encode(all_phrases_kendrick)
 
174
  #grateful_dead_embeddings = encoder_model.encode(all_phrases_grateful_dead)
175
 
176
 
 
130
  messages=messages,
131
  model="google/gemma-2-2b-it",
132
  max_tokens=256,
133
+ temperature=0.85,
134
  #top_p=0.9
135
  )
136
  gemma_response= response["choices"][0]["message"]["content"]
 
170
  radiohead_embeddings = encoder_model.encode(all_phrases_radiohead,
171
  #precision="binary",
172
  show_progress_bar=True,)
173
+ kendrick_embeddings = encoder_model.encode(all_phrases_kendrick,
174
+ show_progress_bar=True,)
175
  #grateful_dead_embeddings = encoder_model.encode(all_phrases_grateful_dead)
176
 
177