Download model during runtime

README.md

@@ -10,8 +10,8 @@ pinned: false
 license: cc-by-4.0
 python_version: 3.12
 short_description: KazSandra project @ ISSAI
-preload_from_hub:
-- "issai/rembert-sentiment-analysis-polarity-classification-kazakh"
+#preload_from_hub:
+#- "issai/rembert-sentiment-analysis-polarity-classification-kazakh"
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -12,3 +12,18 @@ login(os.getenv('HF_TOKEN'))
 hub_folder = Path('~/.cache/huggingface/hub')
 for path in hub_folder.walk():
     st.write(path)
+
+from transformers import AutoModelForSequenceClassification
+from transformers import AutoTokenizer
+from transformers import TextClassificationPipeline
+
+model = AutoModelForSequenceClassification.from_pretrained(
+    "issai/rembert-sentiment-analysis-polarity-classification-kazakh")
+tokenizer = AutoTokenizer.from_pretrained("issai/rembert-sentiment-analysis-polarity-classification-kazakh")
+
+pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer)
+
+reviews = ["Бұл бейнефильм маған түк ұнамады.", "Осы кітап қызық сияқты."]
+
+for review in reviews:
+    st.write(pipe(review))

requirements.txt

@@ -1 +1,2 @@
-streamlit~=1.37.0
+streamlit~=1.37.0
+huggingface-hub~=0.24.5