Update app.py

app.py

@@ -7,27 +7,27 @@ from deoldify import device
 from deoldify.device_id import DeviceId
 from deoldify.visualize import get_image_colorizer
 
-# Ẩn warning khó chịu của fastai
+
 warnings.filterwarnings("ignore", category=UserWarning, message=".*?Your .*? set is empty.*?")
 
-# ⚡ Thiết lập thiết bị
+
 try:
     device.set(device=DeviceId.GPU0)
 except Exception:
     device.set(device=DeviceId.CPU)
 
-# 🧠 Tải model từ Hugging Face Hub (dùng repo có sẵn model)
-REPO_ID = "leonelhs/deoldify"   # 👈 bạn có thể thay repo này bằng repo model của bạn
+
+REPO_ID = "leonelhs/deoldify" 
 snapshot_folder = snapshot_download(repo_id=REPO_ID)
 
-# 🟡 Load DeOldify colorizer
+
 colorizer = get_image_colorizer(root_folder=Path(snapshot_folder), artistic=True)
 
-# 🧾 Hàm dự đoán
+
 def predict(image, render_factor=35):
     return colorizer.get_transformed_image(image, render_factor=render_factor, watermarked=False)
 
-# 🖥️ API/Gradio Interface
+
 demo = gr.Interface(
     fn=predict,
     inputs=[
@@ -35,8 +35,8 @@ demo = gr.Interface(
         gr.Slider(7, 40, value=35, step=1, label="Render Factor (Chất lượng)")
     ],
     outputs=gr.Image(type="pil", label="Ảnh sau khi tô màu"),
-    title="🖼️ DeOldify API - AI Tô Màu Ảnh Trắng Đen",
-    description="Upload ảnh trắng đen — mô hình sẽ tự động tô màu bằng DeOldify.",
+    title="Phục dựng ảnh lịch sử",
+    description="Upload ảnh lịch sử.",
 )
 
 if __name__ == "__main__":