Che237 commited on
Commit
8980f13
·
verified ·
1 Parent(s): 5dd4beb

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -61,7 +61,7 @@ TRAINING_DATA_DIR.mkdir(exist_ok=True)
61
 
62
  # Environment
63
  GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY", "")
64
- GEMINI_MODEL = os.environ.get("GEMINI_MODEL", "gemini-2.0-flash-exp")
65
  HF_TOKEN = os.environ.get("HF_TOKEN", "")
66
  HF_MODEL_REPO = os.environ.get("HF_MODEL_REPO", "Che237/cyberforge-models")
67
 
@@ -793,7 +793,7 @@ async def api_process_dataset(request: Request):
793
  # ============================================================================
794
 
795
  def create_interface():
796
- with gr.Blocks(title="CyberForge AI", theme=gr.themes.Soft()) as demo:
797
  gr.Markdown("""
798
  # 🔐 CyberForge AI - ML Training Platform
799
 
 
61
 
62
  # Environment
63
  GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY", "")
64
+ GEMINI_MODEL = os.environ.get("GEMINI_MODEL", "gemini-2.0-flash")
65
  HF_TOKEN = os.environ.get("HF_TOKEN", "")
66
  HF_MODEL_REPO = os.environ.get("HF_MODEL_REPO", "Che237/cyberforge-models")
67
 
 
793
  # ============================================================================
794
 
795
  def create_interface():
796
+ with gr.Blocks(title="CyberForge AI") as demo:
797
  gr.Markdown("""
798
  # 🔐 CyberForge AI - ML Training Platform
799