pickle-to-skops / app.py
adrin's picture
Update app.py
48936ad
raw
history blame
687 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 = ".".join(in_file.name.split(".")[:-1])
else:
out_file = in_file.name
out_file += ".skops.foo.bar"
_, 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()