Update app.py

app.py

@@ -122,30 +122,44 @@ modelv4 = torch.jit.load(modelarcanev4,map_location='cpu').eval().cpu().half().f
 modelv3 = torch.jit.load(modelarcanev3,map_location='cpu').eval().cpu().half().float()
 modelv2 = torch.jit.load(modelarcanev2,map_location='cpu').eval().cpu().half().float()
 
-def process(im, version):
-    if version == 'version 0.4':
-        im = scale_by_face_size(im, target_face=256, max_res=1_500_000, max_upscale=1)
-        res = proc_pil_img(im, modelv4)
-    elif version == 'version 0.3':
-        im = scale_by_face_size(im, target_face=256, max_res=1_500_000, max_upscale=1)
-        res = proc_pil_img(im, modelv3)
-    else:
-        im = scale_by_face_size(im, target_face=256, max_res=1_500_000, max_upscale=1)
-        res = proc_pil_img(im, modelv2)
+def version4(img):
+    im = scale_by_face_size(im, target_face=256, max_res=1_500_000, max_upscale=1)
+    res = proc_pil_img(im, modelv4)
     return res
-        
-title = "ArcaneGAN"
-description = "Gradio demo for ArcaneGAN, portrait to Arcane style. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below."
-article = "<div style='text-align: center;'>ArcaneGan by <a href='https://twitter.com/devdef' target='_blank'>Alexander S</a> | <a href='https://github.com/Sxela/ArcaneGAN' target='_blank'>Github Repo</a> | <center><img src='https://visitor-badge.glitch.me/badge?page_id=akhaliq_arcanegan' alt='visitor badge'></center></div>"
-
-gr.Interface(
-    process, 
-    [gr.inputs.Image(type="pil", label="Input"),gr.inputs.Radio(choices=['version 0.2','version 0.3','version 0.4'], type="value", default='version 0.4', label='version')
-], 
-    gr.outputs.Image(type="pil", label="Output"),
-    title=title,
-    description=description,
-    article=article,
-    allow_flagging=False,
-    allow_screenshot=False
-    ).launch(enable_queue=True)
+
+def version3(img):
+    im = scale_by_face_size(im, target_face=256, max_res=1_500_000, max_upscale=1)
+    res = proc_pil_img(im, modelv3)
+    return res
+
+def version2(img):
+    im = scale_by_face_size(im, target_face=256, max_res=1_500_000, max_upscale=1)
+    res = proc_pil_img(im, modelv2)
+    return res
+     
+block = gr.Blocks()
+
+with block:
+    gr.Markdown("Gradio Demo for ArcaneGAN, portrait to Arcane style. To use it, simply upload your image. Try out the different versions by clicking on the tabs. Please use a cropped portrait picture for best results.")
+    
+    with gr.Tab("version four"):
+        with gr.Row():
+            facepaint4 = gr.inputs.Image(type="pil",shape=(512,512))
+            faceout4 = gr.outputs.Image(type="pil")
+        face_run = gr.Button("Run")
+        face_run.click(version4, inputs=facepaint4, outputs=faceout4)
+
+    with gr.Tab("version three"):
+        with gr.Row():
+            facepaint3 = gr.inputs.Image(type="pil")
+            faceout3 = gr.outputs.Image(type="pil")
+        face_run = gr.Button("Run")
+        face_run.click(version3, inputs=facepaint3, outputs=faceout3)
+    with gr.Tab("version two"):
+        with gr.Row():
+            facepaint2 = gr.inputs.Image(type="pil")
+            faceout2 = gr.outputs.Image(type="pil")
+        face_run = gr.Button("Run")
+        face_run.click(version2, inputs=facepaint2, outputs=faceout2)
+
+block.launch(enable_queue=True)