File size: 808 Bytes
89f9a8d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
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" |