Upload folder using huggingface_hub

app.py

@@ -566,11 +566,11 @@ def load_model_and_tokenizers():
             # Use the actual vocab sizes and hparams from the loaded config
             model = SmilesIupacLitModule.load_from_checkpoint(
                 checkpoint_path,
-                src_vocab_size=config["actual_src_vocab_size"],
-                tgt_vocab_size=config["actual_tgt_vocab_size"],
+                src_vocab_size=config["src_vocab_size"],
+                tgt_vocab_size=config["tgt_vocab_size"],
                 hparams_dict=config,  # Pass the loaded config as hparams
                 map_location=device,  # Map model to the chosen device (CPU or CUDA)
-                strict=False,  # Be less strict about matching keys, useful for PTL versions or minor changes
+                # strict=False,  # Be less strict about matching keys, useful for PTL versions or minor changes
             )
 
             # Ensure model is on the correct device, in eval mode, and frozen