Update app.py

app.py

@@ -1,29 +1,38 @@
 import gradio as gr
 import uuid
-import shutil
 import os
+from PIL import Image
+import torch
+from evaluate import scan, Detector, seed_torch  # import hàm scan đã sửa
 
 CO_SPY_DIR = "./CO-SPY"
 
+# --- Khởi tạo args và seed trước ---
+class Args:
+    gpu = 0
+    device = f"cuda:{gpu}" if torch.cuda.is_available() else "cpu"
+    semantic_weights_path = "pretrained/semantic_weights.pth"
+    artifact_weights_path = "pretrained/artifact_weights.pth"
+    classifier_weights_path = "pretrained/classifier_weights.pth"
+    batch_size = 32
+    seed = 1024
+
+args = Args()
+seed_torch(args.seed)
+
+# --- Hàm Gradio ---
 def gradio_scan_image(image):
-    # Lưu ảnh upload vào CO-SPY với tên ngẫu nhiên
     dst_path = os.path.join(CO_SPY_DIR, f"input_{uuid.uuid4().hex}.png")
     image.save(dst_path)
-
-    # Import các thư viện cần thiết cho scan
-    from evaluate import scan, args  # giả sử args đã được khởi tạo trong evaluate.py
-
-    # Chạy trực tiếp scan
+    
     probability, label = scan(args, dst_path)
     return f"Prediction: {probability:.3f}, Label: {label}"
 
-# Khởi tạo giao diện Gradio
 iface = gr.Interface(
     fn=gradio_scan_image,
     inputs=gr.Image(type="pil"),
     outputs="text",
-    title="CO-SPY Scan Synthetic Image",
-    description="Upload ảnh và CO-SPY sẽ trả về xác suất và nhãn."
+    title="CO-SPY Scan Synthetic Image"
 )
 
 iface.launch()