Update app.py

app.py

@@ -36,8 +36,8 @@ def run_dif(out_prompt):
                 img_buffer = io.BytesIO(r.content)
                 print (f'bytes:: {io.BytesIO(r.content)}')
                 str_equivalent_image = base64.b64encode(img_buffer.getvalue()).decode()
-                img_tag = "<img class='img_class' src='data:image/png;base64," + str_equivalent_image + "'/>"                
-                out_html+=f"<div><a href='https://huggingface.co/models/{models[i]}'>{models[i]}</a><br>"+img_tag+"</div>"
+                img_tag = "<img src='data:image/png;base64," + str_equivalent_image + "'/>"                
+                out_html+=f"<div  class='img_class'><a href='https://huggingface.co/models/{models[i]}'>{models[i]}</a><br>"+img_tag+"</div>"
                 out = Image.open(io.BytesIO(r.content))
                 out_box.append(out)
             html_out = "<div class='grid_class'>"+out_html+"</div>"
@@ -50,9 +50,10 @@ def run_dif(out_prompt):
 css="""
 .grid_class{
 display:flex;
+height:100%;
 }
 .img_class{
-min-width:100px;
+min-width:200px;
 }
 
 """