Elevi7 commited on
Commit
158ad95
·
verified ·
1 Parent(s): 7aecefc

update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
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(concurrency_count=4, max_size=32)
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()