Try Update app

app.py

@@ -76,6 +76,11 @@ def sample_fn(
     return x_hat.numpy()
 
 
+def sample_with_text_fn(text_prompt):
+    # Not implemented
+    pass
+
+
 def create_videos_and_file(
     gender_radio,
     age_slider,
@@ -171,13 +176,18 @@ def unrest_randomise():
 # TEXT
 title = "Generating Brain Imaging with Diffusion Models"
 description = """
-<center><b>WORK IN PROGRESS. DO NOT SHARE.</b></center>
 <center><a href="https://arxiv.org/">[PAPER]</a> <a href="https://academictorrents.com/details/63aeb864bbe2115ded0aa0d7d36334c026f0660b">[DATASET]</a></center>
 
 <details>
-<summary>Instructions</summary>
-
-With this app, you can generate synthetic brain images with one click!<br />You have two ways to set how your generated brain will look like:<br />- Using the "Inputs" tab that creates well-behaved brains using the same value ranges that our models learned as described in paper linked above<br />- Or using the "Unrestricted Inputs" tab to generate the wildest brains!<br />After customisation, just hit "Generate" and wait a few seconds.<br />Note: if are having problems with the videos, try our app using chrome. <b>Enjoy!<b>
+<summary><b>Instructions</b></summary>
+
+<p style="margin-top: -3px;">With this app, you can generate synthetic brain images with one click!<br />You have several ways to set how your generated brain will look like:<br /></p>
+ <ul style="margin-top: -20px;margin-bottom: -15px;">
+  <li style="margin-bottom: -10px;margin-left: 20px;">Use the "<i>Inputs</i>" tab to create well-behaved brains using the same value ranges that our <br />models learned as described in paper linked above</li>
+  <li style="margin-left: 20px;">Use the "<i>Unrestricted Inputs</i>" tab to generate the wildest brains!</li>
+  <li style="margin-left: 20px;">Use the "<i>Text prompt</i>" tab to generate brains based on text descriptions (Coming soon).</li>
+</ul> 
+<p>After customisation, just hit "<i>Generate</i>" and wait a few seconds.<br />The generated brain will also be available for download, and you can use your favourite Nifti Viewer to check it.<br />Note: if are having problems with the videos, try our app using chrome. <b>Enjoy!<b><p>
 </details>
 
 """
@@ -186,7 +196,8 @@ article = """
 Checkout our dataset with [100K synthetic brain](https://academictorrents.com/details/63aeb864bbe2115ded0aa0d7d36334c026f0660b)! 🧠🧠🧠
 
 App made by [Walter Hugo Lopez Pinaya](https://twitter.com/warvito) from [AMIGO](https://amigos.ai/)
-<center><img src="https://amigos.ai/assets/images/logo_dark_rect.png" alt="amigos.ai" width=300px></center>
+<center><img src="https://raw.githubusercontent.com/Warvito/public_images/master/assets/Footer_1.png" alt="Project by amigos.ai" style="width:450px;"></center>
+<center><img src="https://raw.githubusercontent.com/Warvito/public_images/master/assets/Footer_2.png" alt="Acknowledgements" style="width:750px;"></center>
 """
 
 demo = gr.Blocks()
@@ -280,6 +291,21 @@ with demo:
                                 unrest_brain_number,
                             ],
                         )
+                    with gr.TabItem("Text prompt"):
+                        text_prompt = gr.Textbox("Coming soon... Follow me on twitter to get latest updates.", show_label=False, interactive=False)
+                        submit_text_btn = gr.Button("Generate", variant="primary", )
+                        gr.Examples(
+                            examples=[
+                                ["32 years old | Normal appearance brain"],
+                                ["T2 weighted | Male | 50 years old | There are a few T2 hyperintensities in the deep white matter of the frontal lobes"],
+                                ["Minor small vessel change"],
+                                ["T1 weighted | There is a mild to moderate arachnoid cyst within the anterior left middle cranial fossa"],
+                            ],
+                            inputs=[
+                                text_prompt
+                            ],
+                        )
+
 
         with gr.Column():
             with gr.Box():
@@ -336,5 +362,14 @@ with demo:
         ],
     )
 
+    # submit_text_btn.click(
+    #     fn=sample_with_text_fn,
+    #     inputs=[text_prompt],
+    #     outputs=[axial_sample_plot, sagittal_sample_plot, coronal_sample_plot],
+    # )
+
 # demo.launch(share=True, enable_queue=True)
-demo.launch(enable_queue=True)
+# demo.launch(enable_queue=True)
+demo.queue()
+demo.launch()
+

requirements.txt

@@ -1,5 +1,5 @@
 einops>=0.3.2
-gradio==3.1.1
+gradio==3.3b1
 mediapy==1.0.3
 mlflow
 nibabel