Update app.py
Browse files
app.py
CHANGED
@@ -6,11 +6,11 @@ import pandas as pd
|
|
6 |
import torch
|
7 |
from torch.utils.data import DataLoader
|
8 |
from transformers import AutoTokenizer
|
9 |
-
|
10 |
-
from foldseek_util import get_struc_seq
|
11 |
-
from utils import seed_everything
|
12 |
-
from models import PLTNUM_PreTrainedModel
|
13 |
-
from datasets import PLTNUMDataset
|
14 |
|
15 |
class Config:
|
16 |
batch_size = 2
|
|
|
6 |
import torch
|
7 |
from torch.utils.data import DataLoader
|
8 |
from transformers import AutoTokenizer
|
9 |
+
|
10 |
+
from scripts.foldseek_util import get_struc_seq
|
11 |
+
from scripts.utils import seed_everything
|
12 |
+
from scripts.models import PLTNUM_PreTrainedModel
|
13 |
+
from scripts.datasets import PLTNUMDataset
|
14 |
|
15 |
class Config:
|
16 |
batch_size = 2
|