first app draft

.gitignore

@@ -0,0 +1,3 @@
+.venv/
+tags
+

app.py

@@ -0,0 +1,28 @@
+import streamlit as st
+from huggingface_hub import HfApi, ModelFilter, InferenceClient
+
+if 'models' not in st.session_state:
+
+ hf_api = HfApi()
+ st.session_state.models = sorted(list(hf_api.list_models(
+ filter=ModelFilter(
+ task="text-to-image",
+ library="diffusers",
+ )
+ )), key=lambda x: x.downloads, reverse=True)
+
+def format_model(model):
+ return f"{model.modelId} (⬇️ {model.downloads}, ❤️ {model.likes})"
+
+
+st.title("Text to Image Model testing")
+st.session_state.token = st.text_input("Enter your API token", type="password")
+user_input = st.text_input("Your prompt")
+model_name = st.selectbox("Select", st.session_state.models, format_func=format_model)
+from diffusers import DiffusionPipeline
+
+if model_name and user_input and st.session_state.token:
+ model_name = model_name.modelId
+ client = InferenceClient(model_name, token=st.session_state.token)
+ pred = client.text_to_image(user_input)
+ st.image(pred)

pyproject.toml

@@ -0,0 +1,19 @@
+[tool.poetry]
+name = "images"
+version = "0.1.0"
+description = ""
+authors = ["Philipp Zettl"]
+readme = "README.md"
+
+[tool.poetry.dependencies]
+python = ">=3.10,<4.0"
+streamlit = "^1.28.1"
+requests = "^2.31.0"
+huggingface-hub = "^0.17.0"
+diffusers = "^0.22.1"
+transformers = "^4.35.0"
+
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"

requirements.txt

@@ -0,0 +1,56 @@
+altair==5.1.2 ; python_version >= "3.10" and python_version < "4.0"
+attrs==23.1.0 ; python_version >= "3.10" and python_version < "4.0"
+blinker==1.7.0 ; python_version >= "3.10" and python_version < "4.0"
+cachetools==5.3.2 ; python_version >= "3.10" and python_version < "4.0"
+certifi==2023.7.22 ; python_version >= "3.10" and python_version < "4.0"
+charset-normalizer==3.3.2 ; python_version >= "3.10" and python_version < "4.0"
+click==8.1.7 ; python_version >= "3.10" and python_version < "4.0"
+colorama==0.4.6 ; python_version >= "3.10" and python_version < "4.0" and platform_system == "Windows"
+diffusers==0.22.1 ; python_version >= "3.10" and python_version < "4.0"
+filelock==3.13.1 ; python_version >= "3.10" and python_version < "4.0"
+fsspec==2023.10.0 ; python_version >= "3.10" and python_version < "4.0"
+gitdb==4.0.11 ; python_version >= "3.10" and python_version < "4.0"
+gitpython==3.1.40 ; python_version >= "3.10" and python_version < "4.0"
+huggingface-hub==0.17.3 ; python_version >= "3.10" and python_version < "4.0"
+idna==3.4 ; python_version >= "3.10" and python_version < "4.0"
+importlib-metadata==6.8.0 ; python_version >= "3.10" and python_version < "4.0"
+jinja2==3.1.2 ; python_version >= "3.10" and python_version < "4.0"
+jsonschema-specifications==2023.7.1 ; python_version >= "3.10" and python_version < "4.0"
+jsonschema==4.19.2 ; python_version >= "3.10" and python_version < "4.0"
+markdown-it-py==3.0.0 ; python_version >= "3.10" and python_version < "4.0"
+markupsafe==2.1.3 ; python_version >= "3.10" and python_version < "4.0"
+mdurl==0.1.2 ; python_version >= "3.10" and python_version < "4.0"
+numpy==1.25.2 ; python_version >= "3.10" and python_version < "4.0"
+packaging==23.2 ; python_version >= "3.10" and python_version < "4.0"
+pandas==2.1.2 ; python_version >= "3.10" and python_version < "4.0"
+pillow==10.1.0 ; python_version >= "3.10" and python_version < "4.0"
+protobuf==4.25.0 ; python_version >= "3.10" and python_version < "4.0"
+pyarrow==14.0.0 ; python_version >= "3.10" and python_version < "4.0"
+pydeck==0.8.0 ; python_version >= "3.10" and python_version < "4.0"
+pygments==2.16.1 ; python_version >= "3.10" and python_version < "4.0"
+python-dateutil==2.8.2 ; python_version >= "3.10" and python_version < "4.0"
+pytz==2023.3.post1 ; python_version >= "3.10" and python_version < "4.0"
+pyyaml==6.0.1 ; python_version >= "3.10" and python_version < "4.0"
+referencing==0.30.2 ; python_version >= "3.10" and python_version < "4.0"
+regex==2023.10.3 ; python_version >= "3.10" and python_version < "4.0"
+requests==2.31.0 ; python_version >= "3.10" and python_version < "4.0"
+rich==13.6.0 ; python_version >= "3.10" and python_version < "4.0"
+rpds-py==0.12.0 ; python_version >= "3.10" and python_version < "4.0"
+safetensors==0.4.0 ; python_version >= "3.10" and python_version < "4.0"
+six==1.16.0 ; python_version >= "3.10" and python_version < "4.0"
+smmap==5.0.1 ; python_version >= "3.10" and python_version < "4.0"
+streamlit==1.28.1 ; python_version >= "3.10" and python_version < "4.0"
+tenacity==8.2.3 ; python_version >= "3.10" and python_version < "4.0"
+tokenizers==0.14.1 ; python_version >= "3.10" and python_version < "4.0"
+toml==0.10.2 ; python_version >= "3.10" and python_version < "4.0"
+toolz==0.12.0 ; python_version >= "3.10" and python_version < "4.0"
+tornado==6.3.3 ; python_version >= "3.10" and python_version < "4.0"
+tqdm==4.66.1 ; python_version >= "3.10" and python_version < "4.0"
+transformers==4.35.0 ; python_version >= "3.10" and python_version < "4.0"
+typing-extensions==4.8.0 ; python_version >= "3.10" and python_version < "4.0"
+tzdata==2023.3 ; python_version >= "3.10" and python_version < "4.0"
+tzlocal==5.2 ; python_version >= "3.10" and python_version < "4.0"
+urllib3==2.0.7 ; python_version >= "3.10" and python_version < "4.0"
+validators==0.22.0 ; python_version >= "3.10" and python_version < "4.0"
+watchdog==3.0.0 ; python_version >= "3.10" and python_version < "4.0" and platform_system != "Darwin"
+zipp==3.17.0 ; python_version >= "3.10" and python_version < "4.0"