implementing app
Browse files
app.py
CHANGED
|
@@ -34,7 +34,7 @@ checkpoint_file_atoms = [f for f in dir_list if "ckpt" in f][0]
|
|
| 34 |
model_atom = model_cls.load_from_checkpoint(checkpoint_file_atoms)
|
| 35 |
model_atom.model.roi_heads.score_thresh = 0.65
|
| 36 |
data_cls = Objects_Smiles
|
| 37 |
-
dataset = data_cls(data_path="./uploads/")
|
| 38 |
# dataset.prepare_data()
|
| 39 |
st.title("Atom Level Entity Detector")
|
| 40 |
|
|
|
|
| 34 |
model_atom = model_cls.load_from_checkpoint(checkpoint_file_atoms)
|
| 35 |
model_atom.model.roi_heads.score_thresh = 0.65
|
| 36 |
data_cls = Objects_Smiles
|
| 37 |
+
dataset = data_cls(data_path="./uploads/", batch_size=1)
|
| 38 |
# dataset.prepare_data()
|
| 39 |
st.title("Atom Level Entity Detector")
|
| 40 |
|