Create api-inference.py

api-inference.py

@@ -0,0 +1,6 @@
+from fake_news import main
+
+def query(payload):
+    num_nodes = payload.get("num_nodes", 10)
+    num_links = payload.get("num_links", 5)
+    return main(num_nodes, num_links)