| 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() | |