AI-or-Not-dev

Runtime error

Omnibus commited on Oct 13, 2023

Commit

cba6bca

1 Parent(s): 17da355

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -34,7 +34,7 @@ def aiornot0(image):
     prediction = logits.argmax(-1).item()
     label = labels[prediction]
-    return gr.BarPlot.update(label=f'{models[0]}', vertical=False, x=px, y=[0,1])
 def aiornot1(image):
     labels = ["Real", "AI"]
     mod=models[1]

     prediction = logits.argmax(-1).item()
     label = labels[prediction]
+    return gr.BarPlot.update(label=f'{models[0]}', vertical=False,value=px)
 def aiornot1(image):
     labels = ["Real", "AI"]
     mod=models[1]