Update app.py

app.py

@@ -1,13 +1,14 @@
-# Load model directly
+# Hugging Face 모델 로드
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
-
-tokenizer = AutoTokenizer.from_pretrained("snunlp/KR-FinBert-SC")
-model = AutoModelForSequenceClassification.from_pretrained("snunlp/KR-FinBert-SC")
-
+import streamlit as st
 import os
 import tensorflow as tf
 from absl import logging
 
+# Hugging Face 모델 설정
+tokenizer = AutoTokenizer.from_pretrained("snunlp/KR-FinBert-SC")
+model = AutoModelForSequenceClassification.from_pretrained("snunlp/KR-FinBert-SC")
+
 # 환경 변수 설정
 os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'  # oneDNN 최적화 비활성화
 
@@ -25,61 +26,17 @@ if gpus:
     except RuntimeError as e:
         print(f"GPU 설정 오류: {e}")
 
-# TensorFlow 및 시스템 정보 확인
-print("TensorFlow 버전:", tf.__version__)
-print("사용 가능한 장치:", tf.config.list_physical_devices())
-
-import os
-import tensorflow as tf
-
-# oneDNN 최적화 비활성화
-os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
-
-# GPU 비활성화 (CUDA 문제 해결 시)
-os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
-
-# TensorFlow 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 실행 확인
+# TensorFlow 정보 출력
 print("TensorFlow 버전:", tf.__version__)
 print("사용 가능한 장치:", tf.config.list_physical_devices())
 
-# Base image
-FROM python:3.10-slim
-
-# Install Python packages
-RUN pip install --no-cache-dir pip==22.3.1 \
-    && pip install --no-cache-dir datasets "huggingface-hub>=0.19" \
-    "hf-transfer>=0.1.4" "protobuf<4" "click<8.1" "pydantic~=1.0"
-
-# Install system packages
-RUN apt-get update && apt-get install -y \
-    git git-lfs ffmpeg libsm6 libxext6 cmake rsync libgl1-mesa-glx \
-    && rm -rf /var/lib/apt/lists/*
-
-# Work directory
-WORKDIR /home/user/app
-
-# Copy requirements and install additional Python packages
-COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
-
-# Copy source code
-COPY . .
-
-# Set default command
-CMD ["streamlit", "run", "app.py"]
-
-import streamlit as st
-
+# Streamlit 앱 인터페이스
 st.title("Hello, Streamlit!")
 st.write("This is a sample Streamlit app.")
 
+# 입력 필드 추가
+input_text = st.text_input("Enter some text:")
+if st.button("Analyze"):
+    inputs = tokenizer(input_text, return_tensors="pt")
+    outputs = model(**inputs)
+    st.write("Model Output:", outputs.logits.tolist())