feat: HF_HOME

Dockerfile

@@ -9,10 +9,13 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
-
+ENV PYTHONUNBUFFERED=1 \
+    PORT=8000 \
+    HF_HOME=/home/user/huggingface
+    
 COPY src/ ./src/
 # RUN pip3 install poetry==2.1.3
-# RUN pip install -U "huggingface_hub[cli]"
+# RUN pip install -U "huggingface_hub"
 COPY requirements.txt ./
 RUN pip3 install -r requirements.txt
 # COPY pyproject.toml README.md /app

src/streamlit_app.py

@@ -2,6 +2,7 @@ import os
 import streamlit as st
 import torch
 from transformers import pipeline
+# from huggingface_hub import notebook_login
 # from unsloth import FastLanguageModel, is_bfloat16_supported