Updating with github buttons and gradio version

Signed-off-by: Kieran Fraser <[email protected]>

README.md

@@ -4,7 +4,7 @@ emoji: 🧪
 colorFrom: red
 colorTo: green
 sdk: gradio
-sdk_version: 4.15.0
+sdk_version: 4.16.0
 app_file: app.py
 pinned: false
 license: mit

app.py

@@ -229,9 +229,7 @@ def show_params(type):
         return gr.Column(visible=True)
     return gr.Column(visible=False)  
 
-head = '''
-<script async defer src="https://buttons.github.io/buttons.js"></script>
-'''
+head = f'''<script async defer src="https://buttons.github.io/buttons.js"></script>'''
 
 # e.g. To use a local alternative theme: carbon_theme = Carbon()
 carbon_theme = Carbon()
@@ -259,15 +257,18 @@ with gr.Blocks(css=css, head=head, theme='Tshackelton/IBMPlex-DenseReadable') as
                     <a href="https://github.com/Trusted-AI/adversarial-robustness-toolbox/blob/main/notebooks/hugging_face_poisoning.ipynb" 
                     target="_blank">notebook</a>. Also feel free to contribute and give our repo a ⭐.</i></p>''')
     
-    gr.Markdown('''
-                
-                ''')
+    gr.Markdown('''<div style="width: 100%; text-align: center;">
+                <a style="margin-right: 20px;" class="github-button" 
+                href="https://github.com/Trusted-AI/adversarial-robustness-toolbox" 
+                data-color-scheme="no-preference: light; light: light; dark: dark;" data-size="large" 
+                data-show-count="true" aria-label="Star Trusted-AI/adversarial-robustness-toolbox on GitHub">Star</a> 
+                <!-- Place this tag where you want the button to render. -->
+                <a class="github-button" href="https://github.com/Trusted-AI" 
+                data-color-scheme="no-preference: light; light: light; dark: dark;" data-size="large" data-show-count="true" 
+                aria-label="Follow @Trusted-AI on GitHub">Follow @Trusted-AI</a></div>''')
 
     gr.Markdown('''<hr/>''')
     
-    
-    
-    
     with gr.Row(elem_classes=["larger-gap", "custom-text"]):
         with gr.Column(scale=1):
             gr.Markdown('''<p style="font-size: 20px; text-align: justify">ℹ️ First lets set the scene. You have a dataset of images, such as Imagenette.</p>''')
@@ -328,9 +329,6 @@ with gr.Blocks(css=css, head=head, theme='Tshackelton/IBMPlex-DenseReadable') as
                 outputs=[clean_gallery, poison_gallery, clean_accuracy, poison_success])  
         
     
-        
-        
-    
     gr.Markdown('''<br/>''')
             
 if __name__ == "__main__":
@@ -343,4 +341,4 @@ if __name__ == "__main__":
                 max_threads=20)'''
                 
     # For deployment
-    demo.launch(share=True, ssl_verify=False)
+    demo.launch()

requirements.txt

@@ -7,4 +7,4 @@ tensorflow==2.10.1; sys_platform != "darwin"
 tensorflow-macos; sys_platform == "darwin"
 tensorflow-metal; sys_platform == "darwin"
 adversarial-robustness-toolbox
-gradio==4.15
+gradio==4.16