app.py

import streamlit as st
from .predict_chromosome import predict_and_write


st.title("DeepLoop")

# TODO A drop down of models for different depths
# depth = st.selectbox("Select Model", ["Model 1", "Model 2", "Model 3"])

# Load the model from hugging face
from huggingface_hub import from_pretrained_keras

model = from_pretrained_keras("funlab/DeepLoop-CPGZ-LoopDenoise")


predict_and_write(
    model,
    full_matrix_dir,
    input_name,
    out_dir,
    anchor_dir,
    chromosome,
    small_matrix_size,
    step_size,
    dummy,
    max_dist,
    val_cols,
    keep_zeros,
)