Update app.py

app.py

@@ -10,14 +10,6 @@ import time
 import datetime
 import requests, json
 
-from pprint import pprint
-import hivemind
-from petals.constants import PUBLIC_INITIAL_PEERS
-from health import fetch_health_state
-
-dht = hivemind.DHT(initial_peers=PUBLIC_INITIAL_PEERS, client_mode=True, start=True)
-model_name = "quantumaikr/llama-2-70b-fb16-korean"
-
 loop = asyncio.get_event_loop()
 # Monkey patch
 def get_types(cls_set: List[Type], component: str):
@@ -108,13 +100,12 @@ def get_id_from_p2p(i):
 # Blockchain code
 
 def get_peers(name):
-    data = fetch_health_state(dht)
+    data = requests.get("https://health.petals.dev/api/v1/state").json()
     out = []
     for d in data['model_reports']:
-        if d['name'] == name:
+        if d['name'] == "petals-team/StableBeluga2":
             for r in d['server_rows']:
                 out.append(r['peer_id'])
-
     return out
 
 with gr.Blocks() as demo: