logging

app.py

@@ -1,13 +1,14 @@
 import gradio as gr
 import os
 import random
+import datetime
 from utils import *
 
-file_url = "https://storage.googleapis.com/derendering_model/derendering_supp.zip"
-filename = "derendering_supp.zip"
+# file_url = "https://storage.googleapis.com/derendering_model/derendering_supp.zip"
+# filename = "derendering_supp.zip"
 
-download_file(file_url, filename)
-unzip_file(filename)
+# download_file(file_url, filename)
+# unzip_file(filename)
 print("Downloaded and unzipped the file.")
 
 diagram = get_svg_content("derendering_supp/derender_diagram.svg")
@@ -36,8 +37,17 @@ def demo(Dataset, Model, Output_Format):
     text_outputs = []
     img_outputs = []
     video_outputs = []
-    print("Output format:", Output_Format)
-
+    now = datetime.datetime.now()
+    now = now.strftime("%Y-%m-%d %H:%M:%S")
+    print(
+        now,
+        "Taking sample from dataset:",
+        Dataset,
+        "and model:",
+        Model,
+        "with output format:",
+        Output_Format,
+    )
     for name in picked_samples:
         img_path = os.path.join(path, name)
         img = load_and_pad_img_dir(img_path)