app.py

# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("snunlp/KR-FinBert-SC")
model = AutoModelForSequenceClassification.from_pretrained("snunlp/KR-FinBert-SC")

import os
import tensorflow as tf
from absl import logging

# 환경 변수 설정
os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'  # oneDNN 최적화 비활성화

# 로그 초기화
logging.set_verbosity(logging.INFO)
logging.use_absl_handler()

# GPU 설정
gpus = tf.config.experimental.list_physical_devices('GPU')
if gpus:
    try:
        for gpu in gpus:
            tf.config.experimental.set_memory_growth(gpu, True)
        print("GPU 메모리 증가 허용 설정 완료")
    except RuntimeError as e:
        print(f"GPU 설정 오류: {e}")

# TensorFlow 및 시스템 정보 확인
print("TensorFlow 버전:", tf.__version__)
print("사용 가능한 장치:", tf.config.list_physical_devices())