Spaces:
Sleeping
Sleeping
update app.py
Browse files
app.py
CHANGED
|
@@ -18,7 +18,7 @@ model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
|
|
| 18 |
recent = deque(maxlen=120)
|
| 19 |
|
| 20 |
def render(a):
|
| 21 |
-
ctx = ", ".join(a["context"])
|
| 22 |
return f"**{a['title']}** \n{a['instruction']} \nGoal: {a['goal']} • Duration: {a['duration_min']} min • Energy: {a['energy']} • Context: {ctx}"
|
| 23 |
|
| 24 |
def pick_unique(idxs, goal, energy, max_minutes, ignore_goal=False, ignore_energy=False, ignore_minutes=False, need=3, used=None):
|
|
@@ -54,8 +54,7 @@ def search(query, goal, energy, max_minutes):
|
|
| 54 |
qx = f"{q} Goal:{goal or 'any'} Energy:{energy or 'any'} Max:{int(max_minutes) if max_minutes else ''} minutes"
|
| 55 |
v = model.encode([qx], normalize_embeddings=True)
|
| 56 |
D, I = index.search(np.asarray(v, dtype="float32"), 800)
|
| 57 |
-
idxs = list(I[0])
|
| 58 |
-
random.shuffle(idxs)
|
| 59 |
res, used = [], set()
|
| 60 |
step, used = pick_unique(idxs, goal, energy, max_minutes, False, False, False, 3, used); res += step
|
| 61 |
if len(res) < 3:
|
|
@@ -101,5 +100,5 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo", neutral_hue="slate"))
|
|
| 101 |
cache_examples=False
|
| 102 |
)
|
| 103 |
|
| 104 |
-
demo.queue(
|
| 105 |
demo.launch()
|
|
|
|
| 18 |
recent = deque(maxlen=120)
|
| 19 |
|
| 20 |
def render(a):
|
| 21 |
+
ctx = ", ".join(a["context"]) if isinstance(a.get("context"), list) else (a.get("context") or "")
|
| 22 |
return f"**{a['title']}** \n{a['instruction']} \nGoal: {a['goal']} • Duration: {a['duration_min']} min • Energy: {a['energy']} • Context: {ctx}"
|
| 23 |
|
| 24 |
def pick_unique(idxs, goal, energy, max_minutes, ignore_goal=False, ignore_energy=False, ignore_minutes=False, need=3, used=None):
|
|
|
|
| 54 |
qx = f"{q} Goal:{goal or 'any'} Energy:{energy or 'any'} Max:{int(max_minutes) if max_minutes else ''} minutes"
|
| 55 |
v = model.encode([qx], normalize_embeddings=True)
|
| 56 |
D, I = index.search(np.asarray(v, dtype="float32"), 800)
|
| 57 |
+
idxs = list(I[0]); random.shuffle(idxs)
|
|
|
|
| 58 |
res, used = [], set()
|
| 59 |
step, used = pick_unique(idxs, goal, energy, max_minutes, False, False, False, 3, used); res += step
|
| 60 |
if len(res) < 3:
|
|
|
|
| 100 |
cache_examples=False
|
| 101 |
)
|
| 102 |
|
| 103 |
+
demo.queue()
|
| 104 |
demo.launch()
|