Update app.py

app.py

@@ -29,8 +29,7 @@ def aiornot0(image):
     prediction = logits.argmax(-1).item()
     label = labels[prediction]
     html_out = f"""
-    <h3>
-    <b>This image is likely: {label}</b><br>
+    <h1>This image is likely: {label}</h1><br><h3>
     Model used: <a href='https://huggingface.co/{mod}'>{mod}</a><br>    
     <br>    
     Probabilites:<br>
@@ -51,8 +50,7 @@ def aiornot1(image):
     prediction = logits.argmax(-1).item()
     label = labels[prediction]
     html_out = f"""
-    <h3>
-    <b>This image is likely: {label}</b><br>
+    <h1>This image is likely: {label}</h1><br><h3>
     Model used: <a href='https://huggingface.co/{mod}'>{mod}</a><br>    
     <br>    
     Probabilites:<br>
@@ -73,8 +71,7 @@ def aiornot2(image):
     prediction = logits.argmax(-1).item()
     label = labels[prediction]
     html_out = f"""
-    <h3>
-    <b>This image is likely: {label}</b><br>
+    <h1>This image is likely: {label}</h1><br><h3>
     Model used: <a href='https://huggingface.co/{mod}'>{mod}</a><br>    
     <br>    
     Probabilites:<br>
@@ -82,20 +79,20 @@ def aiornot2(image):
     AI: {px[1][0]}"""
     return gr.HTML.update(html_out)
 with gr.Blocks() as app:
-    with gr.Row():
-        with gr.Column():
-            inp = gr.Image()
-            mod_choose=gr.Number(value=0)
-            btn = gr.Button()
-            
-        with gr.Column():
-            #outp0 = gr.Textbox(label=f'{models[0]}')
-            lab0 = gr.HTML(f"""Testing on Model: {models[0]}""")
-            outp0 = gr.HTML("""""")
-            lab1 = gr.HTML(f"""Testing on Model: {models[1]}""")
-            outp1 = gr.HTML("""""")
-            lab2 = gr.HTML(f"""Testing on Model: {models[2]}""")
-            outp2 = gr.HTML("""""")            
+    with gr.Column():
+        inp = gr.Image()
+        btn = gr.Button()
+    with gr.Group():        
+        with gr.Row():
+            with gr.Box():
+                lab0 = gr.HTML(f"""Testing on Model: {models[0]}""")
+                outp0 = gr.HTML("""""")
+            with gr.Box():
+                lab1 = gr.HTML(f"""Testing on Model: {models[1]}""")
+                outp1 = gr.HTML("""""")
+            with gr.Box():
+                lab2 = gr.HTML(f"""Testing on Model: {models[2]}""")
+                outp2 = gr.HTML("""""")            
     btn.click(aiornot0,[inp],outp0)
     btn.click(aiornot1,[inp],outp1)
     btn.click(aiornot2,[inp],outp2)