|
import json |
|
import os |
|
import pickle |
|
import random |
|
import tarfile |
|
import zipfile |
|
from collections import Counter |
|
from glob import glob |
|
|
|
import gdown |
|
import matplotlib.pyplot as plt |
|
import numpy as np |
|
import pandas as pd |
|
import seaborn as sns |
|
import streamlit as st |
|
from PIL import Image |
|
|
|
import SessionState |
|
|
|
|
|
def download_files( |
|
root_visualization_dir, |
|
viz_url, |
|
viz_archivefile, |
|
demonstration_url, |
|
demonst_zipfile, |
|
picklefile_url, |
|
prediction_root, |
|
prediction_pickle, |
|
): |
|
|
|
if not os.path.exists(root_visualization_dir): |
|
gdown.download(viz_url, viz_archivefile, quiet=False) |
|
os.makedirs(root_visualization_dir, exist_ok=True) |
|
|
|
if viz_archivefile.endswith("tar.gz"): |
|
tar = tarfile.open(viz_archivefile, "r:gz") |
|
tar.extractall(path=root_visualization_dir) |
|
tar.close() |
|
elif viz_archivefile.endswith("zip"): |
|
with zipfile.ZipFile(viz_archivefile, "r") as zip_ref: |
|
zip_ref.extractall(root_visualization_dir) |
|
|
|
|
|
if not os.path.exists(demonst_zipfile): |
|
gdown.download(demonstration_url, demonst_zipfile, quiet=False) |
|
|
|
|
|
with zipfile.ZipFile(demonst_zipfile, "r") as zip_ref: |
|
zip_ref.extractall("./") |
|
|
|
|
|
if not os.path.exists(prediction_pickle): |
|
os.makedirs(prediction_root, exist_ok=True) |
|
gdown.download(picklefile_url, prediction_pickle, quiet=False) |
|
|