Update Dockerfile
Browse files- Dockerfile +1 -11
Dockerfile
CHANGED
@@ -45,17 +45,7 @@ WORKDIR /app
|
|
45 |
RUN chmod -R 755 /app
|
46 |
|
47 |
# Preload SentenceTransformer model with error handling and trust remote code
|
48 |
-
RUN python -c "
|
49 |
-
from sentence_transformers import SentenceTransformer
|
50 |
-
from transformers import AutoModel, AutoTokenizer
|
51 |
-
|
52 |
-
try:
|
53 |
-
model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2', trust_remote_code=True)
|
54 |
-
print('Model loaded successfully')
|
55 |
-
except Exception as e:
|
56 |
-
print(f'Error loading model: {e}')
|
57 |
-
raise
|
58 |
-
"
|
59 |
|
60 |
# Expose port (optional, for web apps)
|
61 |
EXPOSE 7860
|
|
|
45 |
RUN chmod -R 755 /app
|
46 |
|
47 |
# Preload SentenceTransformer model with error handling and trust remote code
|
48 |
+
RUN python -c "from sentence_transformers import SentenceTransformer; from transformers import AutoModel, AutoTokenizer; SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2', trust_remote_code=True)"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
49 |
|
50 |
# Expose port (optional, for web apps)
|
51 |
EXPOSE 7860
|