update

app.py

@@ -94,6 +94,7 @@ def regression_transformer(
                     if "." in s and (s, p) not in correct_samples
                 ]
             )
+        samples = correct_samples
     if task == "Predict":
         return draw_grid_predict(samples[0], target, domain=algorithm.split(":")[0])
     else:

requirements.txt

@@ -8,7 +8,7 @@ torch-sparse
 torch-geometric
 torchvision==0.13.1
 torchaudio==0.12.1
-gt4sd>=1.0.0
+gt4sd>=1.0.6
 molgx>=0.22.0a1
 molecule_generation
 nglview