Upload app.py

app.py

@@ -0,0 +1,29 @@
+import json
+
+import gradio as gr
+import requests
+
+
+def remote_inference(text):
+    url = "http://54.241.102.71:5001/invocations"
+
+    payload = json.dumps({
+        "dataframe_split": {
+            "columns": ["text"],
+            "data": [[text]]
+        }
+    })
+    headers = {
+        'Content-Type': 'application/json'
+    }
+
+    response = requests.request("POST", url, headers=headers, data=payload)
+
+    return {
+        "nodes": response.json()["predictions"][0]["0"],
+        "links": response.json()["predictions"][1]["0"]
+    }
+
+
+iface = gr.Interface(fn=remote_inference, inputs="text", outputs="json")
+iface.launch()