Update Dockerfile to set HF_TOKEN as an environment variable and remove debug print statements from encoder.py for cleaner code.

Dockerfile

@@ -3,6 +3,7 @@ FROM python:3.9
 WORKDIR /app
 
 ENV HF_HOME=/app/hf_cache
+ENV HF_TOKEN=${HF_TOKEN}
 RUN mkdir -p /app/hf_cache && chmod 777 /app/hf_cache
 
 COPY requirements.txt .

src/encoder.py

@@ -1,14 +1,11 @@
 from typing import List, Dict
 from PIL.Image import Image
-import os
 
 import torch
 from transformers import AutoModel, AutoProcessor
 
 
 MODEL_NAME = "Marqo/marqo-fashionCLIP"
-HF_TOKEN = os.environ.get("HF_TOKEN")
-print("token", HF_TOKEN)
 
 
 class FashionCLIPEncoder:
@@ -18,13 +15,11 @@ class FashionCLIPEncoder:
         self.processor = AutoProcessor.from_pretrained(
             MODEL_NAME, 
             trust_remote_code=True,
-            token=HF_TOKEN
         )
         
         self.model = AutoModel.from_pretrained(
             MODEL_NAME,
             trust_remote_code=True,
-            token=HF_TOKEN,
         )
 
         self.model.to(self.device)