clean up. this version worked.

app.py

@@ -17,7 +17,6 @@ HF_TOKEN = os.environ.get("HF_TOKEN") #get HF_TOKEN
 login(token=HF_TOKEN)
 
 
-
 #USER_NAME = os.getenv("USER_NAME", "").strip().strip('"')
 #PRIVATE_SPACE_NAME = os.getenv("PRIVATE_SPACE_NAME", "").strip().strip('"')
 
@@ -26,38 +25,6 @@ login(token=HF_TOKEN)
 REPO_ID ="textbook/textbook_coop"
 REPO_TYPE = "space"
 
-# def setup_cache_directory():
-#     """Set up and return cache directory for private space files"""
-#     cache_dir = Path("private_space_cache")
-#     cache_dir.mkdir(exist_ok=True)
-#     return cache_dir
-
-# def download_private_assets(cache_dir):
-#     """Download necessary files from private space"""
-#     try:
-#         snapshot_download(
-#             repo_id=REPO_ID,
-#             repo_type=REPO_TYPE,
-#             local_dir=cache_dir,
-#         )
-#         return True
-#     except Exception as e:
-#         print(f"Error downloading private assets: {str(e)}")
-#         return False
-
-# # Setup cache directory and download assets
-# cache_dir = setup_cache_directory()
-# download_private_assets(cache_dir)
-
-#Download the entire space (optional, if needed)
-# repo_dir = snapshot_download(
-#     repo_id=REPO_ID,
-#     repo_type=REPO_TYPE,
-#     token=HF_TOKEN,
-#     cache_dir="private_space_cache"
-# )
-
-# # Add repo directory to sys.path so Python can find modules inside it
 # sys.path.append(repo_dir)
 
 # Download the entire space, including the fine-tuned model folder
@@ -69,36 +36,15 @@ repo_dir = snapshot_download(
     force_download=True  # Forces redownload
 )
 
-#print("Files in Model Directory:", os.listdir(repo_dir))
-# Step 2: Change the working directory to the downloaded snapshot directory
-# This ensures that all relative paths inside your private space code are correctly resolved.
-os.chdir(repo_dir)
 
-#print("Files in Model Directory:", os.listdir(repo_dir))
+# Change the working directory to the downloaded snapshot directory
+# This step is very imporptant
+os.chdir(repo_dir)
 
-# Add repo directory to sys.path
+# # Add repo directory to sys.path so Python can find modules inside it
 sys.path.append(repo_dir)
 
 
-
-
-
-
-# Load the base model
-# base_model = AutoModelForCausalLM.from_pretrained(
-#     "meta-llama/Meta-Llama-3-8B",
-#     token=HF_TOKEN_LLAMA
-# )
-
-# # Load fine-tuned adapter (PEFT)
-# fine_tuned_model = PeftModel.from_pretrained(base_model, fine_tuned_model_path)
-
-# # Load tokenizer
-# tokenizer = AutoTokenizer.from_pretrained(fine_tuned_model_path)
-
-# print("Fine-tuned model loaded successfully!")
-
-
 # Download specific files (if snapshot_download wasn't used)
 app_path = hf_hub_download(
     repo_id=REPO_ID,
@@ -112,5 +58,4 @@ app_module = importlib.util.module_from_spec(spec_app)
 spec_app.loader.exec_module(app_module)
 
 # Now you can use functions from utils_module
-result = app_module.main()
-
+result = app_module.main()