Skip to content

Commit

Permalink
Merge pull request #98 from cloudera/ui_sko
Browse files Browse the repository at this point in the history
UI sko
  • Loading branch information
Abhishek-Rnjn authored Feb 6, 2025
2 parents 61b3add + 74d76f7 commit 87ca918
Show file tree
Hide file tree
Showing 5 changed files with 23 additions and 23 deletions.
22 changes: 11 additions & 11 deletions .project-metadata.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -106,25 +106,25 @@ tasks:
memory: 8

- type: start_application
name: Sample-Ticketing-Agent-Application
short_summary: Start Sample Ticketing Agent Application
subdomain: sample-ticketing-agent
script: examples/ticketing-agent-app/ticketing-agent-launch.py
long_summary: This application launcher launches the sample ticketing streamlit application.
name: Fine Tuning Studio
short_summary: Start Fine Tuning Studio
subdomain: fine-tuning-studio
script: bin/run-app.py
long_summary: This application requires an available GPU to run the LLM model and LoRA adapters.
cpu: 2
memory: 8
is_embedded: true
environment_variables:
TASK_TYPE: START_APPLICATION

- type: start_application
name: Fine Tuning Studio
short_summary: Start Fine Tuning Studio
subdomain: fine-tuning-studio
script: bin/run-app.py
long_summary: This application requires an available GPU to run the LLM model and LoRA adapters.
name: Sample-Ticketing-Agent-Application
short_summary: Start Sample Ticketing Agent Application
subdomain: sample-ticketing-agent
script: examples/ticketing-agent-app/ticketing-agent-launch.py
long_summary: This application launcher launches the sample ticketing streamlit application.
cpu: 2
memory: 8
is_embedded: true
environment_variables:
TASK_TYPE: START_APPLICATION
TASK_TYPE: START_APPLICATION
2 changes: 1 addition & 1 deletion bin/download_icons.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
"Navigation": {
"Home": "home",
},
"AI Toolkit": {
"Resources": {
"Import Datasets": "publish",
"View Datasets": "data_object",
"Import Base Models": "neurology",
Expand Down
8 changes: 4 additions & 4 deletions main.py
Original file line number Diff line number Diff line change
Expand Up @@ -153,7 +153,7 @@ def setup_navigation():
</div>
</div>
<div class="dropdown">
<a class="dropbtn" href="#" target="_self"><span class="material-icons">build</span> AI Toolkit</a>
<a class="dropbtn" href="#" target="_self"><span class="material-icons">build</span> Resources</a>
<div class="dropdown-content">
<a href="/datasets" target="_self"><span class="material-icons">publish</span> Import Datasets</a>
<a href="/view_datasets" target="_self"><span class="material-icons">data_object</span> View Datasets</a>
Expand All @@ -174,7 +174,7 @@ def setup_navigation():
</div>
</div>
<div class="dropdown">
<a class="dropbtn" href="#" target="_self"><span class="material-icons">cloud</span> CML</a>
<a class="dropbtn" href="#" target="_self"><span class="material-icons">cloud</span> AI Workbench</a>
<div class="dropdown-content">
<a href="/export" target="_self"><span class="material-icons">upgrade</span> Export And Deploy Model</a>
</div>
Expand Down Expand Up @@ -262,7 +262,7 @@ def setup_sidebar():
st.page_link("pgs/database.py", label="Database Import and Export", icon=":material/database:")
st.write("\n")

st.markdown("AI Toolkit")
st.markdown("Resources")
st.page_link("pgs/datasets.py", label="Import Datasets", icon=":material/publish:")
st.page_link("pgs/view_datasets.py", label="View Datasets", icon=":material/data_object:")
st.page_link("pgs/models.py", label="Import Base Models", icon=":material/neurology:")
Expand All @@ -279,7 +279,7 @@ def setup_sidebar():
st.page_link("pgs/mlflow_jobs.py", label="View MLflow Runs", icon=":material/monitoring:")
st.write("\n")

st.markdown("CML")
st.markdown("AI Workbench")
st.page_link("pgs/export.py", label="Export And Deploy Model", icon=":material/move_group:")
st.write("\n")

Expand Down
4 changes: 2 additions & 2 deletions pgs/export.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,12 +41,12 @@ def create_header():
with col2:
col2.subheader('Export/Deploy Model', divider=DIVIDER_COLOR)
st.caption(
"Export your fine-tuned models and adapters to Cloudera Model Registry or deploy to CML Models for use in production.")
"Export your fine-tuned models and adapters to AI Model Registry or deploy to CAI Models for use in production.")


def handle_export_or_deploy():
with st.container():
deploy_cml_models_tab, export_mr_tab = st.tabs(["**CML Model Deployment**", "**CML Model Registry**"])
deploy_cml_models_tab, export_mr_tab = st.tabs(["**CAI Model Deployment**", "**CAI Model Registry**"])

with deploy_cml_models_tab:
display_model_deployment_tab()
Expand Down
10 changes: 5 additions & 5 deletions pgs/home.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,16 +53,16 @@ def check_amp_update_status():
latest_hash = get_latest_git_hash(current_branch)

with st.container():
st.subheader("πŸ”„ AMP Update Status")
#st.subheader("πŸ”„ AMP Update Status")
if current_hash and latest_hash:
if current_hash != latest_hash:
_, behind = check_if_ahead_or_behind(current_hash, current_branch)
if behind > 0:
st.warning("Your AMP is out of date. Please update to the latest version.")
st.warning("Fine Tuning Studio is out of date. Please update to the latest version.")
if st.button("Perform Upgrade"):
execute_upgrade_steps()
else:
st.success("πŸŽ‰ Your AMP is up-to-date!")
st.success("πŸŽ‰ Fine Tuning Studio is up-to-date!")
else:
st.error("Unable to check AMP update status.")
except Exception as e:
Expand Down Expand Up @@ -122,8 +122,8 @@ def execute_upgrade_steps():
'Finetune your model, leveraging advanced techniques to improve performance.')

create_tile(col4, IconPaths.CML.EXPORT_TO_CML_MODEL_REGISTRY,
"Export to CML Model Registry", "pgs/export.py",
'Export your fine-tuned models and adapters to the Cloudera Model Registry for easy access and deployment.')
"Deploy", "pgs/export.py",
'Export your fine-tuned models and adapters to the AI Model Registry for easy access and deployment.')

st.write("\n")

Expand Down

0 comments on commit 87ca918

Please sign in to comment.