| import sys, signal, base64, re, io, json, time | |
| from io import BytesIO | |
| from pathlib import Path | |
| from typing import Dict | |
| import subprocess | |
| import requests | |
| import gradio as gr | |
| from PIL import Image | |
| import os | |
| from app import MAPPER, save_run | |
| api_key = os.getenv("JINA_TOKEN") | |
| MAPPER.model.set_api_key(api_key) | |
| inputs = [ | |
| ("A person riding a bicycle near the ocean", "https://cdn.duvine.com/wp-content/uploads/2016/04/17095703/Slides_mallorca_FOR-WEB.jpg"), | |
| ("Computing jobs in the US", "https://www.ayresassociates.com/wp-content/uploads/2019/02/Career-Expo-Pie-Chart-Crop.jpg"), | |
| ("Graph of profession choices in bangladesh", "https://notepadacademy.com/wp-content/uploads/2023/08/image.png") | |
| ] | |
| for input in inputs: | |
| img_proc, _, _ = MAPPER.process_image(input[1]) | |
| toks, maps = MAPPER.get_token_similarity_maps(input[0], img_proc) | |
| save_run(input[0], input[1], img_proc, maps) | |