Update app.py

app.py

@@ -16,13 +16,30 @@ for i,model in enumerate(models):
         pass
 print (loaded_model)
 
-def run_dif(out_prompt):
+def run_dif(out_prompt,model_drop,cnt):
     out_box=[]
     out_html=""
-    for i,ea in enumerate(loaded_model):
+    #for i,ea in enumerate(loaded_model):
+    for i in range(cnt):
         try:
-            print (ea)
-            model=loaded_model[int(i)]
+            model=loaded_model[int(model_drop)]
+            out_img=model(out_prompt)
+            print(out_img)
+            out_box.append(out_img)           
+        except Exception as e:
+            print(e)
+            out_html=str(e)
+            pass
+        yield out_box,out_html
+
+def run_dif_og(out_prompt,model_drop,cnt):
+    out_box=[]
+    out_html=""
+    #for i,ea in enumerate(loaded_model):
+    for i in range(cnt):
+        try:
+            #print (ea)
+            model=loaded_model[int(model_drop)]
             out_img=model(out_prompt)
             print(out_img)
             url=f'https://omnibus-top-20.hf.space/file={out_img}'
@@ -69,13 +86,17 @@ def thread_dif(out_prompt,mod):
             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_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)
+        else:
+            out_html=r.status_code
         html_out = "<div class='grid_class'>"+out_html+"</div>"
         return out_box,html_out
     except Exception as e:
-        out_html+=str(e)
+        out_html=str(e)
+        #out_html+=str(e)
         html_out = "<div class='grid_class'>"+out_html+"</div>"
         
         return out_box,html_out
@@ -104,9 +125,14 @@ min-width:200px;
 
 """
 with gr.Blocks(css=css) as app:
-    inp=gr.Textbox()
+    with gr.Row():
+        inp=gr.Textbox(label="Prompt")
+        btn=gr.Button()
+        
+    with gr.Row():
+        model_drop=gr.Dropdown(label="Models", choices=models, type=index, value=models[0])
+        cnt = gr.Number()
     out_html=gr.HTML()
     outp=gr.Gallery()
-    btn=gr.Button()
-    btn.click(run_dif,inp,[outp,out_html])
+    btn.click(run_dif,[inp,model_drop,cnt],[outp,out_html])
 app.launch()