Update app.py

app.py

@@ -5,6 +5,8 @@ from PIL import Image
 import os
 from huggingface_hub import snapshot_download
 
+models = ['DDPM','Glide','Latent Diffusion','Palette','Echt','Stable Diffusion','Unseen Fake','VQ Diffusion']
+
 local_model_path = snapshot_download(repo_id="ElBeh/ma_basemodel")
 model = keras.models.load_model(local_model_path)
 
@@ -12,7 +14,7 @@ def get_prediction(img):
     x = np.array(img)
     x = np.expand_dims(x, axis=0)
     predictions = model.predict(x)
-    return predictions
+    return predictions[0,:]
 
 st.title("Fake Detection")
 
@@ -23,10 +25,8 @@ if file_name is not None:
 
     image = Image.open(file_name)
     col1.image(image, use_column_width=True)  
-    st.subheader('Bild geladen')
-
     predictions = get_prediction(image)
     
     col2.header("Probabilities")
-    for p in predictions:
-        col2.subheader(f"{ p['label'] }: { round(p['score'] * 100, 1)}%")
+    for idx,p in enumerate(predictions):
+        col2.subheader(f"{ idx }: { round(p * 100, 12)}%")