Spaces:
Sleeping
Sleeping
Rob Learsch commited on
Commit ·
a8c9f05
1
Parent(s): 5e55500
Update app.py
Browse files
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=
|
| 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 |
-
|
|
|
|
| 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 |
|