Update app.py

app.py

@@ -110,6 +110,15 @@ def get_peers(model_name):
                 out.append(r['peer_id'])
     return out
 
+
+def get_used_gpus(model_name):
+    data = requests.get("https://health.petals.dev/api/v1/state").json()
+    out = []
+    for d in data['model_reports']:
+        if d['name'] == model_name:
+            return round(d['num_blocks']/0.45, 2)
+    return 0
+
 blockchain = Blockchain()
 
 def add_transaction(id, kind, data):
@@ -148,7 +157,7 @@ def used_gpu():
 
 def get_data():
     output = []
-    output.append({"gpus":get_gpus(), "total":{"total" : blockchain.get_total_gpu_mem(), "used":used_gpu()}, "chain":blockchain.chain})
+    output.append({"gpus":get_gpus(), "total":{"total" : blockchain.get_total_gpu_mem(), "used":get_used_gpus(model_name)}, "chain":blockchain.chain})
 
     return output