Update app.py

app.py

@@ -107,18 +107,16 @@ def get_id_from_p2p(i):
 
 # Blockchain code
 
-def get_peers():
+def get_peers(name):
  data = fetch_health_state(dht)
  out = []
  for d in data['model_reports']:
- if d['name'] == model_name:
+ if d['name'] == name:
  for r in d['server_rows']:
  out.append(r['peer_id'])
 
  return out
 
-get_peers()
-
 with gr.Blocks() as demo:
  count = 0
  aa = gr.Interface(
@@ -183,5 +181,12 @@ with gr.Blocks() as demo:
  outputs="text",
  description="get_p, p2p id로 일반 id 반환. 없으면 no id. \n /run/predict_8",
  )
+
+ gpeer = gr.Interface(
+ fn=get_peers,
+ inputs=["text"],
+ outputs="text",
+ description="get_peers, 해당 모델을 분산처리중인 peer들의 p2p id list 반환\n /run/predict_8",
+ )
  
  demo.queue(max_size=32).launch(enable_queue=True)