Update app.py

app.py

@@ -317,21 +317,34 @@ with gr.Blocks(theme="soft") as demo:
         adv_btn = gr.Button("Download & Compare File")
         adv_result = gr.Textbox(label="Bitwise Comparison Result", lines=3, interactive=False)
 
-        def adv_compare(m1, m2, fname, progress=gr.Progress(track_tqdm=True)):
+        def adv_compare(m1, m2, fname, progress=gr.Progress()):
             if not m1 or not m2 or not fname:
                 return "Please provide both model IDs and the file name."
+            
+            progress(0, desc="Starting download...")
+            
             url1 = f"https://huggingface.co/{m1}/resolve/main/{fname}?download=true"
             url2 = f"https://huggingface.co/{m2}/resolve/main/{fname}?download=true"
+            
             with tempfile.TemporaryDirectory() as tmp:
                 f1 = os.path.join(tmp, "f1.safetensors")
                 f2 = os.path.join(tmp, "f2.safetensors")
+                
+                progress(0.1, desc=f"Downloading from {m1}...")
                 ok1, err1 = download_file_with_progress(url1, f1, progress)
+                
+                progress(0.5, desc=f"Downloading from {m2}...")
                 ok2, err2 = download_file_with_progress(url2, f2, progress)
+                
                 if not ok1 or not ok2:
                     return f"Download error: {err1 or ''} {err2 or ''}"
+                
+                progress(0.9, desc="Comparing files...")
                 percent, err = file_similarity(f1, f2)
                 if err:
                     return f"Comparison error: {err}"
+                
+                progress(1.0, desc="Complete!")
                 return f"Similarity: {percent:.2f}% ({'identical' if percent==100 else 'different'})"
         adv_btn.click(
             fn=adv_compare,