Update app.py

app.py

@@ -17,21 +17,24 @@ print (loaded_model)
 def run_dif(out_prompt):
     out_box=[]
     for i,ea in enumerate(loaded_model):
-        print (ea)
-        model=loaded_model[int(i)]
-        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:
-            out = Image.open(io.BytesIO(r.content))
-            out_box.append(out)
-        yield out_box
-
+        try:
+            print (ea)
+            model=loaded_model[int(i)]
+            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:
+                print (f'bytes:: {io.BytesIO(r.content)}')
+                out = Image.open(io.BytesIO(r.content))
+                out_box.append(out)
+            yield out_box
+        except Exception:
+            yield out_box
 with gr.Blocks() as app:
     inp=gr.Textbox()
     outp=gr.Gallery()