app.py

import os
import sys
import glob  # Needed?
import json  # needed?
import zipfile  # needed?
import numpy as np
import pandas as pd
import streamlit as st
from tqdm import tqdm
from itertools import chain     # needed?

import torch
from torch.utils.data import DataLoader

from rdkit import Chem
from rdkit.Chem import Draw
from rdkit.Chem import AllChem
from rdkit.Chem import DataStructs

sys.path.insert(0, os.path.abspath("src/"))

from clip.clip import _transform
from training.datasets import CellPainting
from clip.model import convert_weights, CLIPGeneral

st.set_page_config(layout="wide")

basepath = os.path.dirname(__file__)
datapath = os.path.join(basepath, "data")

MODEL_PATH = os.path.join(datapath, "epoch_55.pt")

device = "cuda" if torch.cuda.is_available() else "cpu"
model_type = "RN50"