Update app.py

app.py

@@ -4,6 +4,8 @@ from PIL import Image
 import requests
 import uuid
 import io 
+import base64
+
 
 loaded_model=[]
 for i,model in enumerate(models):
@@ -16,6 +18,7 @@ print (loaded_model)
 
 def run_dif(out_prompt):
     out_box=[]
+    out_html=""
     for i,ea in enumerate(loaded_model):
         try:
             print (ea)
@@ -28,16 +31,23 @@ def run_dif(out_prompt):
             #urllib.request.urlretrieve(image, 'tmp.png')
             #out=Image.open('tmp.png')
             r = requests.get(url, stream=True)
+            
             if r.status_code == 200:
+                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 + "'/>"                
+                out_html+=out_html+"<div>"+img_tag+"</div>"
                 out = Image.open(io.BytesIO(r.content))
                 out_box.append(out)
-            yield out_box
+                
+            yield out_box,out_html
         except Exception:
-            yield out_box
+            yield out_box,out_html
 with gr.Blocks() as app:
     inp=gr.Textbox()
+    out_html=gr.HTML()
     outp=gr.Gallery()
     btn=gr.Button()
-    btn.click(run_dif,inp,outp)
+    btn.click(run_dif,inp,[outp,out_html])
 app.launch()