diff --git a/app-check.py b/app-check.py new file mode 100644 index 0000000..eaf4a32 --- /dev/null +++ b/app-check.py @@ -0,0 +1,5 @@ +import streamlit as st +import subprocess + +packages = str(subprocess.run('pip list', capture_output=True)) +st.markdown(packages.replace('\\r\\n', ' \\\n')) diff --git a/polyaxonfiles/app-check.yaml b/polyaxonfiles/app-check.yaml new file mode 100644 index 0000000..fbec76b --- /dev/null +++ b/polyaxonfiles/app-check.yaml @@ -0,0 +1,11 @@ +version: 1.1 +kind: component +name: plx-streamlit-huggingface-app +tags: [streamlit, huggingface] +run: + kind: service + ports: [8501] + rewritePath: true + container: + image: avivkri/polyaxon-examples:plx-streamlit-huggingface-app + command: [streamlit, run, app-check.py] diff --git a/polyaxonfiles/app.yaml b/polyaxonfiles/app.yaml index 20242d6..47b4c68 100644 --- a/polyaxonfiles/app.yaml +++ b/polyaxonfiles/app.yaml @@ -9,4 +9,3 @@ run: container: image: avivkri/polyaxon-examples:plx-streamlit-huggingface-app command: [streamlit, run, app.py] - diff --git a/requirements.txt b/requirements.txt index fb9fb0e..d4d9039 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,4 @@ streamlit==1.25.0 -transformers==4.31.0 +transformers[tf-cpu]==4.31.0 typing-extensions sentencepiece==0.1.99 -tensorflow==2.13.0