File size: 558 Bytes
53448f9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from zipfile import ZipFile
import gradio as gr
def zip_files(files):
with ZipFile("tmp.zip", "w") as zip_obj:
for file in files:
zip_obj.write(file.name, file.name.split("/")[-1])
return "tmp.zip"
demo = gr.Interface(
zip_files,
gr.File(file_count="multiple", file_types=["text", ".json", ".csv"]),
"file",
examples=[[[gr.get_file("titanic.csv"),
gr.get_file("titanic.csv"),
gr.get_file("titanic.csv")]]],
cache_examples=True,
api_name="predict"
)
if __name__ == "__main__":
demo.launch()
|