pickle-to-skops / app.py
adrin's picture
Update app.py
3b41829
raw
history blame
653 Bytes
import gradio as gr
from pathlib import Path
import tempfile
import joblib
from skops import io as sio
def convert(file):
in_file = Path(file.name)
obj = joblib.load(in_file)
if "." in in_file.name:
out_file = in_file.split(".")[:-1]
else:
out_file = in_file
out_file += ".skops"
_, fname = tempfile.mkstemp(suffix=out_file)
sio.dump(obj, fname)
return fname
with gr.Blocks() as iface:
file_output = gr.File()
upload_button = gr.UploadButton("Click to Upload a File", file_types=None, file_count="single")
upload_button.upload(convert, upload_button, file_output)
iface.launch()