Ensure processor is also loaded from Hub

app.py

@@ -18,27 +18,26 @@ warnings.filterwarnings("ignore", category=FutureWarning)
 # --- 모델 및 프로세서 경로 정의 ---
 # Hugging Face Spaces 저장소 내부에 모델 파일을 복사했다고 가정합니다.
 # 저장소 루트에 donut_sroie_finetuned 폴더가 있고 그 안에 final_model 이 있는 구조
-model_path_finetuned = "./donut_sroie_finetuned"
+model_path_finetuned = "greene6517/finetuned_donut_sroie"
 model_name_base = "naver-clova-ix/donut-base" # Base 모델은 Hub에서 직접 로드
 
 # --- Fine-tuned Processor 및 모델 로딩 ---
-# Spaces 환경에서는 우선 local_files_only=True 로 시도합니다 (Git LFS로 파일이 있다면)
-print(f"Loading Fine-tuned processor from: {model_path_finetuned}")
+print(f"Loading Fine-tuned processor from Hub: {model_path_finetuned}") # 로그 메시지도 확인
 try:
-    # local_files_only=True 를 사용하여 Spaces 저장소 내 파일만 사용하도록 강제
-    processor = DonutProcessor.from_pretrained(model_path_finetuned, local_files_only=True)
-    print("Successfully loaded fine-tuned processor locally from Space repo.")
+    # local_files_only=True 가 없어야 함! model_path_finetuned 변수 사용 확인!
+    processor = DonutProcessor.from_pretrained(model_path_finetuned)
+    print("Successfully loaded fine-tuned processor from Hub.")
 except Exception as e:
-    print(f"Error loading fine-tuned processor locally: {e}. Check if model files exist at the path.")
-    # 필요시 Hub에서 로드 시도하는 로직 추가 가능 (단, 모델이 Hub에 업로드 되어 있어야 함)
-    # try:
-    #     processor = DonutProcessor.from_pretrained("your-hf-username/your-model-repo-name") # Hub 경로 예시
-    #     print("Loaded fine-tuned processor from Hub as fallback.")
-    # except Exception as e2:
-    #     print(f"FATAL: Could not load fine-tuned processor locally or from Hub: {e2}")
-    #     exit()
-    # 여기서는 로컬 로딩 실패 시 일단 종료하도록 함 (수정 필요시 주석 해제)
-    print(f"FATAL: Could not load fine-tuned processor locally: {e}")
+    print(f"FATAL: Could not load fine-tuned processor from Hub: {e}")
+    exit()
+
+print(f"Loading Fine-tuned model from Hub: {model_path_finetuned}") # 로그 메시지도 확인
+try:
+    # local_files_only=True 가 없어야 함! model_path_finetuned 변수 사용 확인!
+    model_finetuned = VisionEncoderDecoderModel.from_pretrained(model_path_finetuned)
+    print("Successfully loaded fine-tuned model from Hub.")
+except Exception as e:
+    print(f"FATAL: Could not load fine-tuned model from Hub: {e}")
     exit()
 
 
@@ -298,12 +297,13 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft(primary_hue="blue", secondar
             submit_btn = gr.Button("🚀 결과 비교 시작", variant="primary", scale=0)
             # --- 예제 이미지 부분은 Spaces 환경에서 경로 문제가 있을 수 있어 일단 주석 처리 ---
             # 만약 예제 이미지를 Space 저장소에 함께 업로드하고 경로를 맞출 수 있다면 주석 해제 가능
-            # example_img_dir = "example_images" # 예시: 저장소 루트에 example_images 폴더를 만들고 이미지 넣기
-            # example_paths = [os.path.join(example_img_dir, f) for f in ["example1.jpg", "example2.jpg"] if os.path.exists(os.path.join(example_img_dir, f))]
-            # if example_paths:
-            #     gr.Examples(examples=example_paths, inputs=image_input, label="예제 이미지 클릭 (클릭 후 '결과 비교 시작' 버튼 누르세요)")
-            # else:
-            #     gr.Markdown("_(예제 이미지를 찾을 수 없습니다.)_")
+            example_img_dir = "example" # Space 저장소 루트에 있는 'example' 폴더 지정
+            # list comprehension 사용하여 존재하는 파일만 목록으로 만듦
+            example_paths = [os.path.join(example_img_dir, f) for f in ["1.jpg", "2.jpg"] if os.path.exists(os.path.join(example_img_dir, f))]
+            if example_paths:
+                gr.Examples(examples=example_paths, inputs=image_input, label="예제 이미지 클릭 (클릭 후 '결과 비교 시작' 버튼 누르세요)")
+            else:
+                gr.Markdown("_(예제 이미지를 찾을 수 없습니다. 'example' 폴더 확인 필요)_")
 
         with gr.Column(scale=2):
             with gr.Row(elem_id="output_row"):