Samizie commited on
Commit
f15863b
·
verified ·
1 Parent(s): 68226ff

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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