Update app.py

app.py

@@ -8,6 +8,11 @@ from huggingface_hub import snapshot_download
 local_model_path = snapshot_download(repo_id="ElBeh/ma_basemodel")
 model = keras.models.load_model(local_model_path)
 
+def get_prediction(img):
+    x = image.img_to_array(img)
+    x = np.expand_dims(x, axis=0)
+    predictions = model.predict(x)
+    return predictions
 
 st.title("Fake Detection")
 
@@ -17,12 +22,10 @@ if file_name is not None:
     col1, col2 = st.columns(2)
 
     image = Image.open(file_name)
-    col1.image(image, use_column_width=True)
-    
+    col1.image(image, use_column_width=True)  
     st.subheader('Bild geladen')
-    
-    #predictions = model.predict(image)
-    predictions = 0
+
+    predictions = get_prediction(image)
     
     col2.header("Probabilities")
     for p in predictions: