Spaces:
Sleeping
Sleeping
File size: 788 Bytes
4f48282 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
# config_colab.py
from google.colab import userdata
from config.prompts import prompts # Import prompts from prompts.py
from config.models import models
# Retrieve the Hugging Face token from Colab secrets
api_token = userdata.get("HF_TOKEN")
# Debugging: Check if the Hugging Face token is available
if not api_token:
print("=== Debug: Error ===")
print("ERROR: Hugging Face token (HF_CTB_TOKEN) is missing. Please set it in Colab secrets.")
else:
print("=== Debug: Success ===")
print("Hugging Face token loaded successfully.")
# Debugging: Print prompt and model options
print("=== Debug: Available Options ===")
print("Prompt Options:", [p["alias"] for p in prompts])
print("Model Options:", [m["alias"] for m in models])
print("=================================") |