Update app.py

app.py

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