Update

README.md

@@ -4,7 +4,7 @@ emoji: 🐨
 colorFrom: indigo
 colorTo: red
 sdk: gradio
-sdk_version: 3.36.1
+sdk_version: 4.33.0
 app_file: app.py
 pinned: false
 suggested_hardware: t4-small

app.py

@@ -77,7 +77,10 @@ with gr.Blocks(css="style.css") as demo:
                 text = get_cluster_center_image_markdown(model_name3.value)
                 cluster_center_images = gr.Markdown(value=text)
 
-    model_name.change(fn=model.set_model, inputs=model_name)
+    model_name.change(
+        fn=model.set_model,
+        inputs=model_name,
+    )
     run_button.click(
         fn=model.set_model_and_generate_image,
         inputs=[
@@ -88,7 +91,16 @@ with gr.Blocks(css="style.css") as demo:
         ],
         outputs=result,
     )
-    model_name2.change(fn=get_sample_image_markdown, inputs=model_name2, outputs=sample_images)
-    model_name3.change(fn=get_cluster_center_image_markdown, inputs=model_name3, outputs=cluster_center_images)
+    model_name2.change(
+        fn=get_sample_image_markdown,
+        inputs=model_name2,
+        outputs=sample_images,
+    )
+    model_name3.change(
+        fn=get_cluster_center_image_markdown,
+        inputs=model_name3,
+        outputs=cluster_center_images,
+    )
 
-demo.queue(max_size=10).launch()
+if __name__ == "__main__":
+    demo.queue(max_size=10).launch()

requirements.txt

@@ -1,6 +1,6 @@
 lpips==0.1.4
-numpy==1.23.5
-Pillow==10.0.0
-scipy==1.10.1
+numpy==1.26.4
+Pillow==10.3.0
+scipy==1.13.1
 torch==2.0.1
 torchvision==0.15.2

style.css

@@ -1,11 +1,8 @@
 h1 {
   text-align: center;
+  display: block;
 }
 div#result {
   max-width: 600px;
   max-height: 600px;
 }
-img#visitor-badge {
-  display: block;
-  margin: auto;
-}