Update app.py

app.py

@@ -33,8 +33,16 @@ def aiornot0(image):
         print(px)
         
     prediction = logits.argmax(-1).item()
-    label = labels[prediction] 
-    return gr.BarPlot.update(label=f'{models[0]}', vertical=False,value=px, x=px,y=labels)
+    label = labels[prediction]
+    
+    html_out = f"""
+    <h3>Model used: <a href src='https://huggingface.co/models/{mod}'>{mod}</a><br>
+    This image is likely: {label}<br>
+    Probabilites<br>
+    AI: {px[0]}<br>
+    Real: {px[1]}"""
+    
+    return gr.update(html_out)
 def aiornot1(image):    
     labels = ["Real", "AI"]
     mod=models[1]
@@ -72,7 +80,7 @@ with gr.Blocks() as app:
             
         with gr.Column():
             #outp0 = gr.Textbox(label=f'{models[0]}')
-            outp0 = gr.BarPlot(label=f'{models[0]}', vertical=False)
+            outp0 = gr.HTML("""""")
             outp1 = gr.Textbox(label=f'{models[1]}')
             outp2 = gr.Textbox(label=f'{models[2]}')
     btn.click(aiornot0,[inp],outp0)