Update app.py

app.py

@@ -68,9 +68,8 @@ with gr.Blocks(js=js,theme=gr.themes.Soft()) as demo:
         #image_tensor = image_tensor / 255.0
         
         model = load_model('densenet121_model.keras')
-        #prediction = model.predict(img)[0]
-        prediction = model.predict(img).flatten()
-
+        prediction = model.predict(img)[0]
+       
         predicted_class = {class_names[i]: float(prediction[i]) for i in range(4)}
         return predicted_class