Update app.py

app.py

@@ -28,12 +28,12 @@ def aiornot0(image):
         print (logits)
         probability = softmax(logits)
         print(f'PROBABILITY ::: {probability}')
-        print(probability[0])
+        print(probability[0][0])
         px = pd.DataFrame(probability.numpy())
         
     prediction = logits.argmax(-1).item()
     label = labels[prediction] 
-    return px
+    return gr.BarPlot(label=f'{models[0]}', vertical=False, x=px, y=[0,1])
 def aiornot1(image):    
     labels = ["Real", "AI"]
     mod=models[1]