Spaces:
Running
Running
Upload app.py with huggingface_hub
Browse files
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
|
| 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"
|
| 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 |
|