kootaeng2 commited on
Commit
a8e057a
ยท
1 Parent(s): 3372c29

Finalize: Correct Dockerfile CMD and __init__ structure

Browse files
Files changed (3) hide show
  1. Dockerfile +5 -0
  2. force_upload.py +27 -0
  3. src/__init__.py +4 -1
Dockerfile CHANGED
@@ -11,6 +11,11 @@ RUN pip install --no-cache-dir -r requirements.txt
11
  # 4. ํ”„๋กœ์ ํŠธ ์ „์ฒด ์ฝ”๋“œ ๋ณต์‚ฌ
12
  COPY . .
13
 
 
 
 
 
 
14
  # 5. Hugging Face Spaces๊ฐ€ ์‚ฌ์šฉํ•  ํฌํŠธ(7860) ์—ด๊ธฐ
15
  EXPOSE 7860
16
 
 
11
  # 4. ํ”„๋กœ์ ํŠธ ์ „์ฒด ์ฝ”๋“œ ๋ณต์‚ฌ
12
  COPY . .
13
 
14
+ # Hugging Face ๊ด€๋ จ ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ๋“ค์ด ์‚ฌ์šฉํ•  ์บ์‹œ ํด๋”๋ฅผ
15
+ # ๊ถŒํ•œ์ด ์žˆ๋Š” /app ํด๋” ๋‚ด๋ถ€๋กœ ์ง€์ •ํ•ฉ๋‹ˆ๋‹ค.
16
+ ENV HF_HOME=/app/.cache
17
+ ENV TRANSFORMERS_CACHE=/app/.cache
18
+
19
  # 5. Hugging Face Spaces๊ฐ€ ์‚ฌ์šฉํ•  ํฌํŠธ(7860) ์—ด๊ธฐ
20
  EXPOSE 7860
21
 
force_upload.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # force_upload.py
2
+ from huggingface_hub import HfApi
3
+ import os
4
+
5
+ # --- ์„ค์ • ---
6
+ # PC์— ์žˆ๋Š”, ์˜ฌ๋ฐ”๋ฅด๊ฒŒ ์ˆ˜์ •๋œ ๋ชจ๋ธ ํด๋” ๊ฒฝ๋กœ
7
+ LOCAL_MODEL_DIR = "./korean-emotion-classifier-final"
8
+ # ๋ฎ์–ด์“ฐ๊ธฐ ํ•  Hub์˜ ๋ชจ๋ธ ID
9
+ REPO_ID = "koons/korean-emotion-classifier-final"
10
+ # --- --- ---
11
+
12
+ # Hugging Face API ํด๋ผ์ด์–ธํŠธ ์ƒ์„ฑ
13
+ api = HfApi()
14
+
15
+ # ํด๋” ์—…๋กœ๋“œ
16
+ print(f"'{LOCAL_MODEL_DIR}' ํด๋”์˜ ๋‚ด์šฉ์„ '{REPO_ID}'์— ์—…๋กœ๋“œํ•ฉ๋‹ˆ๋‹ค...")
17
+ try:
18
+ # folder_path์— ์žˆ๋Š” ๋ชจ๋“  ํŒŒ์ผ์„ repo_id์— ์˜ฌ๋ฆฝ๋‹ˆ๋‹ค. (๊ธฐ์กด ํŒŒ์ผ์€ ๋ฎ์–ด์”€)
19
+ api.upload_folder(
20
+ folder_path=LOCAL_MODEL_DIR,
21
+ repo_id=REPO_ID,
22
+ repo_type="model"
23
+ )
24
+ print("โœ… ์—…๋กœ๋“œ ์™„๋ฃŒ!")
25
+ except Exception as e:
26
+ print(f"\nโŒ ์—…๋กœ๋“œ ์ค‘ ์˜ค๋ฅ˜๊ฐ€ ๋ฐœ์ƒํ–ˆ์Šต๋‹ˆ๋‹ค: {e}")
27
+ print("huggingface-cli login์ด ์ •์ƒ์ ์œผ๋กœ ๋˜์—ˆ๋Š”์ง€, REPO_ID๊ฐ€ ์ •ํ™•ํ•œ์ง€ ํ™•์ธํ•ด์ฃผ์„ธ์š”.")
src/__init__.py CHANGED
@@ -16,8 +16,11 @@ def create_app():
16
  db.init_app(app)
17
 
18
  from . import main, auth
19
- app. register_blueprint(main.bp)
20
  app.register_blueprint(auth.bp)
21
 
22
 
 
 
 
23
  return app
 
16
  db.init_app(app)
17
 
18
  from . import main, auth
19
+ app.register_blueprint(main.bp)
20
  app.register_blueprint(auth.bp)
21
 
22
 
23
+ with app.app_context():
24
+ from . import models
25
+ db
26
  return app