update

app.py

@@ -123,7 +123,7 @@ def load_models_correct_architecture(checkpoint_path):
 
     try:
         # Load checkpoint
-        checkpoint = torch.load(checkpoint_path, map_location=device, weights_only=False)
+        checkpoint = torch.load(checkpoint_path, map_location=device)
         print("🔍 Checkpoint structure:")
         print(f"Keys: {list(checkpoint.keys())}")
 
@@ -261,8 +261,12 @@ def generate_emoji_fixed(input_image):
 print("✅ Main inference function ready!")
 
 def gradio_wrapper(input_image):
-    result = generate_emoji_fixed(input_image)
-    return result
+    img = generate_emoji_fixed(input_image)
+    if isinstance(img, tuple):
+        # ignore status message, chỉ return image
+        img = img[0]
+    return img
+
 
 if __name__ == "__main__":
     # try:
@@ -295,5 +299,6 @@ if __name__ == "__main__":
 
     # Launch
     print("🌐 Launching Fixed DTN Demo...")
-    port = int(os.environ.get("PORT", 7860))
-    demo.launch(server_name="0.0.0.0", server_port=port, share=False, debug=False)
+    # port = int(os.environ.get("PORT", 7860))
+    # demo.launch(server_name="0.0.0.0", server_port=port, share=False, debug=False)
+    demo.launch(debug=True)