Update app.py

app.py

@@ -72,12 +72,13 @@ if file_name is not None:
     col1.image(image, use_column_width=True)  
     predictions = get_prediction(image)
     
-    col2.header("Probabilities")
     if binary:
-        if predictions[7] > 0.5:
+        col2.header("Prediction")
+        if predictions[6] > 0.5:
             col2.markdown(":green[real image!]")
         else:
             col2.markdown(":red[fake image!]")
     else:
+        col2.header("Probabilities")
         for idx,p in enumerate(predictions):
             col2.text(f"{ models[idx] }: { round(p * 100, 2)}%")