Nicer border and static hosted images

app.py

@@ -132,15 +132,18 @@ if __name__ == "__main__":
                 with gr.Row():
                     with gr.Column():
                         gr.Markdown(f"### Base Prompt")
-                        gr.Image(value=f"{example['base']['src']}")
+                        #gr.Image(value=f"{example['base']['src']}")
+                        gr.HTML(f"<img src='https://huggingface.co/spaces/treadon/prompt-fungineer-355M/resolve/main/{example['base']['src']}' style='width: 100%; border-radius: 15px; border: 1px solid #444' />")
                         gr.Markdown(f"{example['base']['prompt']}")
                     with gr.Column():
                         gr.Markdown(f"### 355M Prompt Fungineered")
-                        gr.Image(value=f"{example['355M']['src']}")
+                        #gr.Image(value=f"{example['355M']['src']}")
+                        gr.HTML(f"<img src='https://huggingface.co/spaces/treadon/prompt-fungineer-355M/resolve/main/{example['355M']['src']}' style='width: 100%; border-radius: 15px; border: 1px solid #444' />")
                         gr.Markdown(f"{example['355M']['prompt']}")
                     with gr.Column():
                         gr.Markdown(f"### 7B Prompt Fungineered")
-                        gr.Image(value=f"{example['7B']['src']}")
+                        #gr.Image(value=f"{example['7B']['src']}")
+                        gr.HTML(f"<img src='https://huggingface.co/spaces/treadon/prompt-fungineer-355M/resolve/main/{example['7B']['src']}' style='width: 100%; border-radius: 15px; border: 1px solid #444' />")
                         gr.Markdown(f"{example['7B']['prompt']}")