Update app.py

app.py

@@ -7,8 +7,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, LogitsProcessor, L
 
 
 # === Model Configuration ===
-MODEL_ID = os.getenv("MODEL_ID", "llm-rna-api-rmit/rna-structure-model")
-HF_TOKEN = os.getenv("HF_TOKEN", None)
+MODEL_ID = os.getenv("MODEL_ID", "llm-rna-api-rmit/rna-structure-model") #Model can be changed but it may not work without modification to eval code
+HF_TOKEN = os.getenv("HF_TOKEN", None) #Currently not required as model is public
 TOKENIZER = None
 MODEL = None
 VALID_CHARS = re.compile(r"^[AUCG]+$")
@@ -304,6 +304,7 @@ def _precompute_can_open(seq, min_loop=3, allow_gu=True):
 
 
 # === Balanced Parentheses Processor ===
+# dot_bias , paren_penalty and window may be changed in order to modify the models bias's to get it more accurate
 class BalancedParenProcessor(LogitsProcessor):
     """Custom logits processor that ensures balanced parentheses during structure generation."""
     def __init__(self, lp_id, rp_id, dot_id, total_len, can_open,
@@ -372,7 +373,7 @@ def _top_p_sample(logits, top_p=0.5, temperature=0.4):
     idx = torch.multinomial(sorted_probs, 1)
     return sorted_idx.gather(-1, idx).squeeze(-1)
 
-
+# top_p, temperature and min_loop can be changed in order to change model biases
 def _generate_db(seq, top_p=0.5, temperature=0.4, min_loop=3, greedy=False):
     if TOKENIZER is None or MODEL is None:
         raise RuntimeError("Model not initialized — call init_model() first.")