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)