Update app.py

app.py

@@ -1,28 +1,18 @@
 import os,sys
+from huggingface_hub import HfApi
+
+# Hugging Face 토큰 설정
+HF_TOKEN = "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"  # 실제 토큰으로 교체
 
 # install required packages
-os.system('pip install python-dotenv')  # dotenv 설치
-os.system('pip install plotly')
-os.system('pip install matplotlib')
+os.system('pip install -q transformers')
+os.system('pip install -q plotly')
+os.system('pip install -q matplotlib')
 os.system('pip install dgl==1.0.2+cu116 -f https://data.dgl.ai/wheels/cu116/repo.html')
 os.environ["DGLBACKEND"] = "pytorch"
 print('Modules installed')
 
-from transformers import pipeline
-from datasets import load_dataset
-import gradio as gr
-from dotenv import load_dotenv
-
-# 환경 변수 로드
-load_dotenv()
-HF_TOKEN = os.getenv("HF_TOKEN")
-if not HF_TOKEN:
-    print("Warning: HF_TOKEN not found in environment variables. Using default settings.")
-    HF_TOKEN = "YOUR_DEFAULT_TOKEN"  # 기본값 설정 또는 에러 처리
-
-
-
-# 기존 import문들 유지
+# 기존 import문들
 import plotly.graph_objects as go
 import numpy as np
 import py3Dmol
@@ -36,7 +26,7 @@ from utils.parsers_inference import parse_pdb
 from model.util import writepdb
 from utils.inpainting_util import *
 
-# 새로운 모델과 데이터셋 로드
+# LLM과 데이터셋 로드
 pipe = pipeline("text-generation", 
                model="CohereForAI/c4ai-command-r-plus-08-2024",
                token=HF_TOKEN)