Update app.py

src/app.py

@@ -1,6 +1,11 @@
 import os
 from pathlib import Path
 
+# -----------------
+# Get the directory where app.py is located
+# -----------------
+APP_DIR = Path(__file__).parent.resolve()
+
 # -----------------
 # Fix Streamlit Permission Issues
 # -----------------
@@ -80,7 +85,7 @@ def load_model(from_pretrained, base_model, cache_dir, device):
 # Configuration
 # -----------------
 MODEL_CONFIG = {
-    'from_pretrained': './FineTune/ckpt/lr_0.0001_a_1',
+    'from_pretrained': str(APP_DIR / 'FineTune' / 'ckpt' / 'lr_0.0001_a_1'),
     'base_model': 'gemma-1b',
     'cache_dir': '../cache', 
     'device': 'mps',