shekkari21 commited on
Commit
add2495
Β·
1 Parent(s): 3bc689a

modified files

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -6
  2. app.py +8 -10
Dockerfile CHANGED
@@ -2,15 +2,11 @@
2
  # you will also find guides on how best to write your Dockerfile
3
 
4
  FROM python:3.9
5
- RUN --mount=type=secret,id=CLEARML_API_ACCESS_KEY,required=true \
6
- echo "Using ClearML API key: $(cat /run/secrets/CLEARML_API_ACCESS_KEY)" && \
7
- pip install clearml
8
  RUN useradd -m -u 1000 user
9
  USER user
10
  ENV PATH="/home/user/.local/bin:$PATH"
11
- ENV CLEARML_WEB_HOST="https://app.clear.ml/"
12
- ENV CLEARML_API_HOST="https://api.clear.ml"
13
- ENV CLEARML_FILES_HOST="https://files.clear.ml"
14
  WORKDIR /app
15
 
16
  COPY --chown=user ./requirements.txt requirements.txt
 
2
  # you will also find guides on how best to write your Dockerfile
3
 
4
  FROM python:3.9
5
+
 
 
6
  RUN useradd -m -u 1000 user
7
  USER user
8
  ENV PATH="/home/user/.local/bin:$PATH"
9
+
 
 
10
  WORKDIR /app
11
 
12
  COPY --chown=user ./requirements.txt requirements.txt
app.py CHANGED
@@ -11,24 +11,22 @@ import argparse
11
  from argparse import Namespace
12
  from peft import PeftModel, PeftConfig, get_peft_model, LoraConfig
13
 
14
- # ── Load ClearML secrets from HF Spaces environment ───────────────────────────
15
- CLEARML_WEB_SERVER = os.environ["CLEARML_WEB_SERVER"]
16
- CLEARML_API_SERVER = os.environ["CLEARML_API_SERVER"]
17
- CLEARML_FILES_SERVER = os.environ["CLEARML_FILES_SERVER"]
18
  CLEARML_ACCESS_KEY = os.environ["CLEARML_API_ACCESS_KEY"]
19
  CLEARML_SECRET_KEY = os.environ["CLEARML_API_SECRET_KEY"]
20
- # ───────────────────────────────────────────────────────────────────────────────
21
 
22
- # Configure ClearML SDK with injected environment variables
23
  Task.set_credentials(
24
- api_host=CLEARML_API_SERVER,
25
- web_host=CLEARML_WEB_SERVER,
26
- files_host=CLEARML_FILES_SERVER,
27
  key=CLEARML_ACCESS_KEY,
28
  secret=CLEARML_SECRET_KEY,
29
  )
30
 
31
- MAX_SOURCE_LENGTH = 512
32
 
33
  def pad_assert(tokenizer, source_ids):
34
  source_ids = source_ids[:MAX_SOURCE_LENGTH - 2]
 
11
  from argparse import Namespace
12
  from peft import PeftModel, PeftConfig, get_peft_model, LoraConfig
13
 
14
+ # Load endpoints & creds
15
+ CLEARML_API_HOST = os.environ["CLEARML_API_HOST"]
16
+ CLEARML_WEB_HOST = os.environ["CLEARML_WEB_HOST"]
17
+ CLEARML_FILES_HOST = os.environ["CLEARML_FILES_HOST"]
18
  CLEARML_ACCESS_KEY = os.environ["CLEARML_API_ACCESS_KEY"]
19
  CLEARML_SECRET_KEY = os.environ["CLEARML_API_SECRET_KEY"]
 
20
 
21
+ # Apply to SDK
22
  Task.set_credentials(
23
+ api_host=CLEARML_API_HOST,
24
+ web_host=CLEARML_WEB_HOST,
25
+ files_host=CLEARML_FILES_HOST,
26
  key=CLEARML_ACCESS_KEY,
27
  secret=CLEARML_SECRET_KEY,
28
  )
29
 
 
30
 
31
  def pad_assert(tokenizer, source_ids):
32
  source_ids = source_ids[:MAX_SOURCE_LENGTH - 2]