Update app.py

app.py

@@ -2,4 +2,29 @@
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
 tokenizer = AutoTokenizer.from_pretrained("snunlp/KR-FinBert-SC")
-model = AutoModelForSequenceClassification.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())