Update app.py

app.py

@@ -62,11 +62,14 @@ def convert_and_push(radio_model_names, input_model, ckpt_name, token):
         )
 
         # 3. Push to the model repo
+        commit_message="Add Diffusers weights"
         upload_folder(
             folder_path=model_id,
             repo_id=model_id,
             token=token,
             create_pr=True,
+            commit_message=commit_message,
+            commit_description=f"Add Diffusers weights converted from checkpoint {ckpt_name} in revision {revision}",
         )
 
         # # 4. Delete the downloaded checkpoint file, yaml files, and the converted model folder
@@ -76,14 +79,20 @@ def convert_and_push(radio_model_names, input_model, ckpt_name, token):
         for f in glob.glob("*.yaml*"):
             subprocess.run(["rm", "-rf", f])
 
-        return "Success"
+        return f"""Successfully converted the checkpoint and opened a PR to add the weights to the model repo.
+                You can view and merge the PR [here]({hf_utils.get_pr_url(HfApi(token=token), model_id, commit_message)})."""
     
     except Exception as e:
         return error_str(e)
 
-    
+
+DESCRIPTION = """### Convert a stable diffusion checkpoint to Diffusers🧨
+                With this space, you can easily convert a CompVis stable diffusion checkpoint to Diffusers and automatically create a pull request to the model repo.
+                You can choose to convert a checkpoint from one of your models, or from any other model on the Hub."""
+
 with gr.Blocks() as demo:
 
+    gr.Markdown(DESCRIPTION)
     with gr.Row():
 
         with gr.Column(scale=11):
@@ -92,9 +101,9 @@ with gr.Blocks() as demo:
                 input_token = gr.Textbox(
                     max_lines=1,
                     label="Enter your Hugging Face token",
-                    placeholder="hf_...",
+                    placeholder="READ permission is enough",
                 )
-                gr.Markdown("You can get a token [here](https://huggingface.co/settings/tokens).")
+                gr.Markdown("You can get a token [here](https://huggingface.co/settings/tokens)")
                 with gr.Group(visible=False) as group_model:
                     radio_model_names = gr.Radio(label="Choose a model")
                     input_model = gr.Textbox(
@@ -144,5 +153,11 @@ with gr.Blocks() as demo:
         scroll_to_output=True
     )
 
+    # gr.Markdown("""<img src="https://raw.githubusercontent.com/huggingface/diffusers/main/docs/source/imgs/diffusers_library.jpg" width="150"/>""")
+    gr.HTML("""
+    <p>Space by: <a href="https://twitter.com/hahahahohohe"><img src="https://img.shields.io/twitter/follow/hahahahohohe?label=%40anzorq&style=social" alt="Twitter Follow"></a></p><br>
+    <p><img src="https://visitor-badge.glitch.me/badge?page_id=anzorq.sd-to-diffusers" alt="visitors"></p>
+    """)
+    
 demo.queue()
 demo.launch()