Spaces:
Runtime error
Runtime error
Commit
·
56cba7c
1
Parent(s):
5ca8659
update utils
Browse files
utils.py
CHANGED
@@ -26,10 +26,10 @@ def generate_workout(system_prompt, query, knowledge):
|
|
26 |
return response
|
27 |
|
28 |
|
29 |
-
def run(gender, muscle_group, equipment, level, duration, vectorstore, system_prompt, k=5, randomize=True):
|
30 |
query = f"{duration}-minute {muscle_group} workout for {gender} {level} level {equipment}"
|
31 |
knowledge = retrieve_knowledge(query, vectorstore, k, randomize)
|
32 |
-
response = generate_workout(system_prompt, query, knowledge)
|
33 |
|
34 |
for i in range(len(response)):
|
35 |
time.sleep(0.01)
|
|
|
26 |
return response
|
27 |
|
28 |
|
29 |
+
def run(gender, muscle_group, equipment, level, duration, vectorstore, system_prompt, llm, k=5, randomize=True):
|
30 |
query = f"{duration}-minute {muscle_group} workout for {gender} {level} level {equipment}"
|
31 |
knowledge = retrieve_knowledge(query, vectorstore, k, randomize)
|
32 |
+
response = generate_workout(system_prompt, query, knowledge, llm)
|
33 |
|
34 |
for i in range(len(response)):
|
35 |
time.sleep(0.01)
|