| import requests | |
| import gradio as gr | |
| def infer(im): | |
| im.save("converted.png") | |
| url = "https://ajax.thehive.ai/api/demo/classify?endpoint=text_recognition" | |
| files = { | |
| "image": ("converted.png", open("converted.png", "rb"), "image/png"), | |
| "model_type": (None, "detection"), | |
| "media_type": (None, "photo"), | |
| } | |
| headers = {"referer": "https://thehive.ai/"} | |
| res = requests.post(url, headers=headers, files=files) | |
| text = "" | |
| blocks = [] | |
| for output in res.json()["response"]["output"]: | |
| text += output["block_text"] | |
| for poly in output["bounding_poly"]: | |
| blocks.append({ | |
| "text": "".join([c["class"] for c in poly["classes"]]), | |
| "rect": poly["dimensions"] | |
| }) | |
| return text | |
| iface = gr.Interface( | |
| fn=infer, | |
| title="Mariam - Beta", | |
| description=" La maj du siècle. ", | |
| inputs=[gr.Image(type="pil")], | |
| outputs=["text"], | |
| ).launch() |