Update app.py
Browse files
app.py
CHANGED
@@ -528,7 +528,7 @@ def load_model_and_tokenizers():
|
|
528 |
|
529 |
|
530 |
# --- Inference Function for Gradio ---
|
531 |
-
|
532 |
def predict_iupac(smiles_string, decoding_strategy, num_beams, num_return_sequences):
|
533 |
"""
|
534 |
Performs SMILES to IUPAC translation using the loaded model and selected strategy.
|
|
|
528 |
|
529 |
|
530 |
# --- Inference Function for Gradio ---
|
531 |
+
@spaces.GPU # Uncomment if using GPU and have appropriate hardware tier
|
532 |
def predict_iupac(smiles_string, decoding_strategy, num_beams, num_return_sequences):
|
533 |
"""
|
534 |
Performs SMILES to IUPAC translation using the loaded model and selected strategy.
|