import os

MODEL_NAME = "deepseek-ai/deepseek-llm-r1-7b"
HF_TOKEN = os.getenv("HF_TOKEN")  # Set this in the Hugging Face environment
DEVICE = "cuda"  # Change to "cpu" if no GPU