Update app.py

app.py

@@ -1,19 +1,11 @@
 import tensorflow as tf
 import numpy as np
-from tensorflow.keras.models import load_model
-from tensorflow.keras.utils import CustomObjectScope
-from PIL import Image
 from transformers import TFAutoModel
+from PIL import Image
 import gradio as gr
 
-def absolute_difference(x):
-    return tf.abs(x[0] - x[1])
-    
-model_path = "huggingface_Nithikorn/Signature_Detection"
-
-# ตรวจสอบว่าไฟล์อยู่ที่นี่จริง
-with CustomObjectScope({'absolute_difference': absolute_difference}):
-    model = load_model(model_path)
+# โหลดโมเดลจาก Hugging Face
+model = TFAutoModel.from_pretrained("huggingface_Nithikorn/Signature_Detection")
 print("✅ โมเดลโหลดสำเร็จ!")
 
 def preprocess_image(image):
@@ -26,7 +18,7 @@ def preprocess_image(image):
 # ฟังก์ชันเปรียบเทียบลายเซ็น
 def predict_signature(image1, image2):
     print("เริ่มการทำนาย...")
-    
+
     if image1 is None or image2 is None:
         print("ไม่พบรูปภาพ")
         return "❌ กรุณาอัปโหลดภาพทั้งสองภาพ"