Update app.py

app.py

@@ -8,7 +8,7 @@ from urllib.parse import urlparse
 import xml.etree.ElementTree as ET
 
 # 모델 경로와 디바이스 설정
-model_path = r'ssocean/NAIP'
+model_path = 'ssocean/NAIP'
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 # 전역 변수로 모델·토크나이저 선언
@@ -16,7 +16,7 @@ model = None
 tokenizer = None
 
 def fetch_arxiv_paper(arxiv_input):
-    """arXiv URL 또는 ID로부터 제목과 요약(fetch)"""
+    """arXiv URL 또는 ID로부터 제목과 요약 가져오기"""
     try:
         if 'arxiv.org' in arxiv_input:
             parsed = urlparse(arxiv_input)
@@ -48,9 +48,10 @@ def predict(title, abstract):
         model = AutoModelForSequenceClassification.from_pretrained(
             model_path,
             num_labels=1,
-            quantization_config=None,   # bitsandbytes 양자화 완전 비활성화
-            torch_dtype=torch.float32,  # 전부 float32
-            device_map=None,            # accelerate dispatch 비활성
+            torch_dtype=torch.float32,   # 전체 float32
+            device_map=None,             # 자동 분산 비활성화
+            load_in_8bit=False,
+            load_in_4bit=False,
             low_cpu_mem_usage=False
         )
         tokenizer = AutoTokenizer.from_pretrained(model_path)
@@ -139,7 +140,7 @@ css = """
     background: white;
     padding: 2rem;
     border-radius: 1rem;
-    box-shadow: 0 4px 6px -1px rgb(0 0 0 / 0.1);
+    box-shadow: 0 4px 6px -1px rgba(0,0,0,0.1);
 }
 .result-section {
     background: #f8fafc;