Update app.py

app.py

@@ -37,15 +37,23 @@ def run_dif(out_prompt):
                 print (f'bytes:: {io.BytesIO(r.content)}')
                 str_equivalent_image = base64.b64encode(img_buffer.getvalue()).decode()
                 img_tag = "<img src='data:image/png;base64," + str_equivalent_image + "'/>"                
-                out_html+=out_html+"<div>"+img_tag+"</div>"
+                out_html+=f"<div><a href='https://huggingface.co/models/{ea}'>{ea}</a><br>"+img_tag+"</div>"
                 out = Image.open(io.BytesIO(r.content))
                 out_box.append(out)
-                
-            yield out_box,out_html
+            html_out = "<div class='grid_class'>"+out_html+"</div>"
+            yield out_box,html_out
         except Exception as e:
             out_html+=e
-            yield out_box,out_html
-with gr.Blocks() as app:
+            html_out = "<div class='grid_class'>"+out_html+"</div>"
+            
+            yield out_box,html_out
+css="""
+.grid_class{
+display:flex;
+}
+
+"""
+with gr.Blocks(css=css) as app:
     inp=gr.Textbox()
     out_html=gr.HTML()
     outp=gr.Gallery()