Update app.py

app.py

@@ -83,16 +83,18 @@ def gradio_interface(base_model_name, chat_model_name, hf_token, load_one_at_a_t
     layer_diffs = calculate_layer_diffs(base_model, chat_model, load_one_at_a_time=load_one_at_a_time)
     return visualize_layer_diffs(layer_diffs, base_model_name, chat_model_name)
 
-iface = gr.Interface(
-    fn=gradio_interface,
-    inputs=[
-        gr.Textbox(lines=2, placeholder="Enter base model name"),
-        gr.Textbox(lines=2, placeholder="Enter chat model name"),
-        gr.Textbox(lines=2, placeholder="Enter Hugging Face token", type="password"),  # Hide token input
-        gr.Checkbox(label="Load one layer at a time")
-    ],
-    outputs=gr.Image(type="pil"),  # Specify PIL image output
-    title="Model Weight Difference Visualizer"
-)
-
-iface.launch()
+if __name__ == "__main__":
+    iface = gr.Interface(
+        fn=gradio_interface,
+        inputs=[
+            gr.Textbox(label="Base Model Name", lines=2),
+            gr.Textbox(label="Chat Model Name", lines=2),
+            gr.Textbox(label="Hugging Face Token", type="password", lines=2),
+            gr.Checkbox(label="Load one layer at a time")
+        ],
+        outputs=gr.Image(type="pil", label="Weight Differences Visualization"),
+        title="Model Weight Difference Visualizer",
+        cache_examples=False
+    )
+    
+    iface.launch(share=False, server_port=7860)