Update app.py

app.py

@@ -1,7 +1,10 @@
 import os
 os.environ["HF_HOME"] = "/tmp/hf_cache"
 os.makedirs("/tmp/hf_cache", exist_ok=True)
-
+from huggingface_hub import whoami    
+os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
+import sys
+import spaces
 from fastapi import FastAPI, Query
 from huggingface_hub import list_repo_files, hf_hub_download, upload_file
 import io
@@ -19,8 +22,21 @@ import os
 import os
 import zipfile
 import tempfile  # ✅ Add this!
+import yaml
+sys.path.insert(0, os.getcwd())
 
+import gradio as gr
+from PIL import Image
+import torch
+import uuid
+import os
+import shutil
+import json
+import yaml
+from slugify import slugify
+from transformers import AutoProcessor, AutoModelForCausalLM
 
+sys.path.insert(0, "ai-toolkit")
 
 
 app = FastAPI()
@@ -430,6 +446,7 @@ def start_training(
         push_to_hub = False
 
     slugged_lora_name = lora_name.replace(" ", "_").lower()
+    print(username)
 
     # Load base config
     config = {