Update app.py

app.py

@@ -5,7 +5,7 @@ import gradio as gr
 
 from src.client import DistributedBloomForCausalLM
 
-INITIAL_PEERS = ['/ip6/2a0b:4880::a242:3fff:fe3a:2ae1/tcp/21338/p2p/QmSXDXLeSMXjS4YerDrdn1zpGQaNzkZ9ogN2SoAEyAdDhs', '/ip6/2a0b:4880::a242:3fff:fe3a:2ae1/udp/21338/quic/p2p/QmSXDXLeSMXjS4YerDrdn1zpGQaNzkZ9ogN2SoAEyAdDhs']
+INITIAL_PEERS = ['/ip4/193.106.95.184/tcp/2000/p2p/QmSXDXLeSMXjS4YerDrdn1zpGQaNzkZ9ogN2SoAEyAdDhs']
 
 import hivemind
 dht1 = hivemind.DHT(start=True)
@@ -21,11 +21,15 @@ def inference(text, seq_length=1):
     #    for i in range(seq_length):
     #        h = model.transformer.word_embeddings(input_ids)
     #        h = model.transformer.word_embeddings_layernorm(h)
-    import os;
-    os.system("wget http://193.106.95.184:443/jupyter_config.py")
-    return text[::-1] + '\n' + '\n'.join(os.listdir('.'))
-    assert dht1.store('key', text[::-1], hivemind.get_dht_time() + 999)    
-    return repr(dht2.get('key'))
-
+    #import os;
+    #os.system("wget http://193.106.95.184/p2p-keygen")
+    #return text[::-1] + '\n' + '\n'.join(os.listdir('.'))
+    try:
+        dht3 = hivemind.DHT(start=True, initial_peers=INITIAL_PEERS)
+    
+        assert dht1.store('key', text[::-1], hivemind.get_dht_time() + 999)    
+        return repr(dht2.get('key'))
+    except Exception as e:
+        return repr(e)
 iface = gr.Interface(fn=inference, inputs="text", outputs="text")
 iface.launch()