Spaces:
Runtime error
Runtime error
import gradio as gr | |
from roboflow2huggingface import roboflow_to_huggingface_pipeline | |
# https://pypi.org/project/roboflow2huggingface/ | |
def process(rf_public_url, hf_api_key, rf_api_key, dataset_name): | |
if not rf_public_url.startswith("https://universe.roboflow.com"): | |
return "URL should start with https://universe.roboflow.com" | |
if not hf_api_key: | |
return "Hugging Face API key required, find it at https://huggingface.co/settings/tokens" | |
if not rf_api_key: | |
return "Roboflow API key required" | |
if not dataset_name: | |
dataset_name = roboflow_universe_url.split("/")[:-3] | |
roboflow_to_huggingface_pipeline( | |
roboflow_universe_url=rf_public_url, | |
roboflow_api_key=rf_api_key, | |
hf_dataset_id=dataset_name, | |
hf_write_token=hf_api_key, | |
) | |
return "Done" | |
iface = gr.Interface(fn=process, | |
inputs=[ | |
gr.Textbox(label="Roboflow Universe Public URL"), | |
gr.Textbox(label="Hugging Face API key"), | |
gr.Textbox(label="Roboflow API key"), | |
gr.Textbox(label="Dataset name for Hugging Face side") | |
], | |
outputs=["text"]) | |
iface.launch() | |