Rulga commited on
Commit
038fc2b
·
1 Parent(s): f33b7bd

Refactor model configuration to use ACTIVE_MODEL for improved clarity and consistency

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -273,7 +273,8 @@ with gr.Blocks() as demo:
273
  with gr.Column(scale=3):
274
  chatbot = gr.Chatbot(
275
  label="Chat",
276
- avatar_images=None
 
277
  )
278
 
279
  with gr.Row():
@@ -383,18 +384,18 @@ with gr.Blocks() as demo:
383
  gr.Markdown(f"""
384
  **Base Model Path:**
385
  ```
386
- {MODEL_CONFIG['training']['base_model_path']}
387
  ```
388
 
389
  **Fine-tuned Model Path:**
390
  ```
391
- {MODEL_CONFIG['training']['fine_tuned_path']}
392
  ```
393
 
394
  **LoRA Configuration:**
395
- - Rank (r): {MODEL_CONFIG['training']['lora_config']['r']}
396
- - Alpha: {MODEL_CONFIG['training']['lora_config']['lora_alpha']}
397
- - Dropout: {MODEL_CONFIG['training']['lora_config']['lora_dropout']}
398
  """)
399
 
400
  with gr.Tab("Model Training"):
 
273
  with gr.Column(scale=3):
274
  chatbot = gr.Chatbot(
275
  label="Chat",
276
+ avatar_images=None,
277
+ type='messages'
278
  )
279
 
280
  with gr.Row():
 
384
  gr.Markdown(f"""
385
  **Base Model Path:**
386
  ```
387
+ {ACTIVE_MODEL['training']['base_model_path']}
388
  ```
389
 
390
  **Fine-tuned Model Path:**
391
  ```
392
+ {ACTIVE_MODEL['training']['fine_tuned_path']}
393
  ```
394
 
395
  **LoRA Configuration:**
396
+ - Rank (r): {ACTIVE_MODEL['training']['lora_config']['r']}
397
+ - Alpha: {ACTIVE_MODEL['training']['lora_config']['lora_alpha']}
398
+ - Dropout: {ACTIVE_MODEL['training']['lora_config']['lora_dropout']}
399
  """)
400
 
401
  with gr.Tab("Model Training"):