File size: 610 Bytes
96410e5
59e2445
bc25885
af7831d
59e2445
96410e5
6cb74a2
59e2445
 
 
6cb74a2
59e2445
 
 
96410e5
59e2445
 
05873ed
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# config.py
import os
from config.prompts import prompts  # Import prompts from prompts.py
from config.models import models

# Retrieve the Hugging Face token
api_token = os.getenv("HF_TOKEN")

# Debugging: Check if the Hugging Face token is available
if not api_token:
    print("ERROR: Hugging Face token (HF_TOKEN) is missing. Please set it as an environment variable or in Colab secrets.")
else:
    print("Hugging Face token loaded successfully.")


# Debugging: Print prompt and model options
print("Prompt Options:", [p["alias"] for p in prompts])
print("Model Options:", [m["alias"] for m in models])