Spaces:
Running
Running
File size: 391 Bytes
4f48282 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
# config.py
import os
from config.prompts import prompts # Direct Import prompts from prompts.py
from config.models import models # Direct Import models
# Retrieve the Hugging Face token
api_token = os.getenv("HF_TOKEN")
# Debugging: Print prompt and model options
print("Prompt Options:", [p["alias"] for p in prompts])
print("Model Options:", [m["alias"] for m in models])
gpu = "T4"
|