Update app.py

app.py

@@ -37,7 +37,7 @@ 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+=f"<div><a href='https://huggingface.co/models/{ea}'>{ea}</a><br>"+img_tag+"</div>"
+                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>"