deployed model

.gitignore

@@ -1,3 +1,7 @@
 .DS_Store
 backend/.DS_Store
-backend/models/bert-mood-prediction-1.pt
+backend/models/bert-mood-prediction-1.pt
+config.json
+backend/models/.locks
+backend/models/models--dhruthick--my-bert-lyrics-classifier
+hf_cache

Dockerfile

@@ -1,15 +1,19 @@
-FROM python
+FROM python:3.9
 
+# The two following lines are requirements for the Dev Mode to be functional
+# Learn more about the Dev Mode at https://huggingface.co/dev-mode-explorers
+RUN useradd -m -u 1000 user
 WORKDIR /app
 
-# Copy the requirements.txt file into the image
-COPY requirements.txt .
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
-# Install the Python dependencies
-RUN pip install --upgrade pip
-RUN pip install -r requirements.txt
+COPY --chown=user . /app
 
-COPY . .
+USER user
+
+ENV HOME=/home/user \
+	PATH=/home/user/.local/bin:$PATH
 
 EXPOSE 7860
 

app.py

@@ -3,8 +3,12 @@ from lyricsgenius import Genius
 import json
 import torch
 import numpy as np
+import os
 from transformers import BertTokenizer, BertForSequenceClassification, AutoTokenizer, AutoModelForSequenceClassification
 
+# Set the TRANSFORMERS_CACHE environment variable
+os.environ['TRANSFORMERS_CACHE'] = './hf_cache'
+
 app = Flask(__name__)
 
 mood_map = {