Extract CALIBRATION_FILE

app.py

@@ -71,6 +71,7 @@ class HuggingFaceModelProcessor:
     ERROR_LOGIN = "You must be logged in to use GGUF-my-repo."
     DOWNLOAD_FOLDER = "./downloads"
     OUTPUT_FOLDER = "./outputs"
+    CALIBRATION_FILE = "calibration_data_v5_rc.txt"
 
     def __init__(self):
         self.SPACE_ID = os.environ.get("SPACE_ID", "")
@@ -139,7 +140,7 @@ class HuggingFaceModelProcessor:
         if not os.path.isfile(quant_config.fp16_model):
             raise GGUFConverterError(f"Model file not found: {quant_config.fp16_model}")
 
-        train_data_path = "calibration_data_v5_rc.txt"
+        train_data_path = CALIBRATION_FILE
         if not os.path.isfile(train_data_path):
             raise GGUFConverterError(f"Training data file not found: {train_data_path}")
         print(f"Training data file path: {train_data_path}")