Update app.py

app.py

@@ -36,14 +36,14 @@ 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 src='data:image/png;base64," + str_equivalent_image + "'/>"                
+                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>"
                 out = Image.open(io.BytesIO(r.content))
                 out_box.append(out)
             html_out = "<div class='grid_class'>"+out_html+"</div>"
             yield out_box,html_out
         except Exception as e:
-            out_html+=e
+            out_html+=str(e)
             html_out = "<div class='grid_class'>"+out_html+"</div>"
             
             yield out_box,html_out
@@ -51,6 +51,9 @@ css="""
 .grid_class{
 display:flex;
 }
+.img_class{
+min-width:100px;
+}
 
 """
 with gr.Blocks(css=css) as app: