Update app.py

app.py

@@ -13,6 +13,10 @@ import yaml
 with open('config.yaml', 'r') as file:
     config = yaml.safe_load(file)
 
+# Debug: Print the structure of the config
+print("Config structure:")
+print(yaml.dump(config, default_flow_style=False))
+
 # Authenticate using the token stored in Hugging Face Spaces secrets
 if 'HF_TOKEN' in os.environ:
     login(token=os.environ['HF_TOKEN'])
@@ -20,10 +24,22 @@ else:
     raise ValueError("HF_TOKEN not found in environment variables. Please add it to your Space's secrets.")
 
 # Correctly access the config values
-base_model = config['config']['model']['name_or_path']
+try:
+    base_model = config['config']['model']['name_or_path']
+except KeyError:
+    print("Could not find 'config' -> 'model' -> 'name_or_path' in the config file.")
+    print("Available keys in config:", list(config.keys()))
+    if 'config' in config:
+        print("Available keys in config['config']:", list(config['config'].keys()))
+    base_model = "FLUX.1-dev"  # Default value
+
 lora_model = "sagar007/sagar_flux"  # This isn't in the config, so we're keeping it as is
-trigger_word = config['config']['trigger_word']
 
+try:
+    trigger_word = config['config']['trigger_word']
+except KeyError:
+    print("Could not find 'config' -> 'trigger_word' in the config file.")
+    trigger_word = "sagar" 
 # Global variables
 pipe = None
 cache = {}