Update app.py

app.py

@@ -2,7 +2,8 @@ import gradio as gr
 import torch
 from transformers import AutoFeatureExtractor, AutoModelForImageClassification, pipeline
 from numpy import exp
- 
+import pandas as  pd
+
 def softmax(vector):
  e = exp(vector)
  return e / e.sum()
@@ -27,10 +28,12 @@ def aiornot0(image):
         print (logits)
         probability = softmax(logits)
         print(f'PROBABILITY ::: {probability}')
-
+        print(probability[0])
+        px = pd.DataFrame(probability.numpy())
+        
     prediction = logits.argmax(-1).item()
     label = labels[prediction] 
-    return label
+    return px
 def aiornot1(image):    
     labels = ["Real", "AI"]
     mod=models[1]
@@ -67,7 +70,8 @@ with gr.Blocks() as app:
             btn = gr.Button()
             
         with gr.Column():
-            outp0 = gr.Textbox(label=f'{models[0]}')
+            #outp0 = gr.Textbox(label=f'{models[0]}')
+            outp0 = gr.BarPlot(label=f'{models[0]}', vertical=False)
             outp1 = gr.Textbox(label=f'{models[1]}')
             outp2 = gr.Textbox(label=f'{models[2]}')
     btn.click(aiornot0,[inp],outp0)