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.
|