from env

app.py

@@ -2,7 +2,6 @@ import os
 import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
-from keys import HF_TOKEN
 
 # Set cache directory for HF Spaces persistent storage
 os.environ.setdefault("HF_HOME", "/data/.huggingface")
@@ -17,8 +16,7 @@ model_list = {
 
 # Use token from environment variables (HF Spaces) or keys.py (local)
 HF_TOKEN_FROM_ENV = os.getenv("HUGGINGFACEHUB_API_TOKEN") or os.getenv("HF_TOKEN")
-if HF_TOKEN_FROM_ENV:
-    HF_TOKEN = HF_TOKEN_FROM_ENV
+HF_TOKEN = HF_TOKEN_FROM_ENV
 
 # Model cache for loaded models
 model_cache = {}