import DEFAULT_WEIGHT from fsrs_optimizer

app.py

@@ -5,7 +5,7 @@ import sys
 if os.environ.get("DEV_MODE"):
     # for local development
     sys.path.insert(0, os.path.abspath("../fsrs-optimizer/src/fsrs_optimizer/"))
-from fsrs_optimizer import Optimizer
+from fsrs_optimizer import Optimizer, DEFAULT_WEIGHT
 
 
 def interface_func(weights: str, ratings: str, request_retention: float) -> str:
@@ -25,7 +25,7 @@ iface = gr.Interface(
         gr.inputs.Textbox(
             label="weights",
             lines=1,
-            default="0.4, 0.9, 2.3, 10.9, 4.93, 0.94, 0.86, 0.01, 1.49, 0.14, 0.94, 2.18, 0.05, 0.34, 1.26, 0.29, 2.61",
+            default=str(DEFAULT_WEIGHT)[1:-1],
         ),
         gr.inputs.Textbox(label="ratings", lines=1, default="3,3,3,3,1,3,3"),
         gr.inputs.Slider(