Update app.py

app.py

@@ -5,7 +5,7 @@ import requests
 import uuid
 import io 
 import base64
-
+import threading
 
 loaded_model=[]
 for i,model in enumerate(models):
@@ -16,7 +16,7 @@ for i,model in enumerate(models):
         pass
 print (loaded_model)
 
-def run_dif(out_prompt):
+def run_dif(out_prompt,mod):
     out_box=[]
     out_html=""
     for i,ea in enumerate(loaded_model):
@@ -47,6 +47,49 @@ def run_dif(out_prompt):
             html_out = "<div class='grid_class'>"+out_html+"</div>"
             
             yield out_box,html_out
+
+def thread_dif(out_prompt,mod):
+    out_box=[]
+    out_html=""
+    #for i,ea in enumerate(loaded_model):
+    try:
+        print (ea)
+        model=loaded_model[int(mod)]
+        out_img=model(out_prompt)
+        print(out_img)
+        url=f'https://omnibus-top-20.hf.space/file={out_img}'
+        print(url)
+        uid = uuid.uuid4()
+        #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+=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>"
+        yield out_box,html_out
+    except Exception as e:
+        out_html+=str(e)
+        html_out = "<div class='grid_class'>"+out_html+"</div>"
+        
+        yield out_box,html_out
+
+
+            
+def start_threads(prompt):
+    t1 = threading.Thread(target=thread_dif, args=(prompt,0))
+    t2 = threading.Thread(target=thread_dif, args=(prompt,1))
+    t1.start()
+    t2.start()
+    print (t1)
+    print (t2)
+            
 css="""
 .grid_class{
 display:flex;
@@ -62,5 +105,5 @@ with gr.Blocks(css=css) as app:
     out_html=gr.HTML()
     outp=gr.Gallery()
     btn=gr.Button()
-    btn.click(run_dif,inp,[outp,out_html])
+    btn.click(start_threads,inp,[outp,out_html])
 app.launch()