Update app.py

app.py

@@ -13,6 +13,38 @@ loaded_model=[]
 for i,model in enumerate(models):
     loaded_model.append(gr.load(f'models/{model}', cache_examples=False))
 
+
+
+
+
+
+
+
+def get_concat_h_cut(in1, in2):
+    im1=Image.open(in1)
+    im2=Image.open(in2)
+    dst = Image.new('RGB', (im1.width + im2.width,
+                            min(im1.height, im2.height)))
+    dst.paste(im1, (0, 0))
+    dst.paste(im2, (im1.width, 0))
+    return dst
+
+
+def get_concat_v_cut(in1,theme='light'):
+    im1=Image.open(in1)
+    if theme=='dark':
+        color=(31,41,55)
+    if theme=='light':
+        color=(255,255,255)
+    dst = Image.new('RGB', (min(im1.width, im2.width), im1.height + im2.height+100),color=color)
+    dst.paste(im1, (0, 0))
+    return dst
+
+
+
+
+
+
 def format_prompt(message, history):
   prompt = "<s>"
   for user_prompt, bot_response in history:
@@ -69,6 +101,7 @@ def run(inp,model_drop):
     r = requests.get(url, stream=True)
     if r.status_code == 200:
         out = Image.open(io.BytesIO(r.content))
+    out=get_concat_v_cut(out)
     return out
 
 def run_gpt(in_prompt,history,):