minor fixes

config.py

@@ -16,6 +16,9 @@
 
 DOMAIN = os.getenv("DOMAIN")
 
+REPLICATE_OWNER = "josebenitezg"
+HF_OWNER = "joselobenitezg"
+
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)

main.py

@@ -7,7 +7,6 @@
 from gradio_app import login_demo, main_demo
 import gradio as gr
 from pathlib import Path
-from fastapi.middleware.cors import CORSMiddleware
 
 app = FastAPI()
 
@@ -16,7 +15,6 @@
 
 static_dir = Path("./static")
 app.mount("/static", StaticFiles(directory=static_dir, html=True), name="static")
-#app.mount("/assets", StaticFiles(directory="assets", html=True), name="assets")
 
 app.add_middleware(SessionMiddleware, secret_key=SECRET_KEY, max_age=3600)
 

services/train_lora.py

@@ -2,13 +2,12 @@
 import os
 from huggingface_hub import create_repo
 from database import create_lora_models
-
-REPLICATE_OWNER = "josebenitezg"
+from config import REPLICATE_OWNER, HF_OWNER
 
 def lora_pipeline(user_id, zip_path, model_name, trigger_word="TOK", steps=1000, lora_rank=16, batch_size=1, autocaption=True, learning_rate=0.0004):
     print(f'Creating dataset for {model_name}')
     model_name = model_name.lower().replace(' ', '_')
-    hf_repo_name = f"joselobenitezg/flux-dev-{model_name}"
+    hf_repo_name = f"{HF_OWNER}/flux-dev-{model_name}"
     replicate_repo_name = f"josebenitezg/flux-dev-{model_name}"
     create_repo(hf_repo_name, repo_type='model')
 
@@ -45,7 +44,6 @@ def lora_pipeline(user_id, zip_path, model_name, trigger_word="TOK", steps=1000,
         destination=f"{model.owner}/{model.name}"
     )
 
-    print(f"training: {training.keys()}")
     print(f"Training started: {training.status}")
     print(f"Training URL: https://replicate.com/p/{training.id}")
     print(f"Creating model in Database")