diff --git a/mlaas/__pycache__/__init__.cpython-310.pyc b/mlaas/__pycache__/__init__.cpython-310.pyc
index 9f0fedc1fda0b8b2d47ac0822b44c09467c3ce37..a169a85ef514e207929ef211680d2f8629ff8fac 100644
Binary files a/mlaas/__pycache__/__init__.cpython-310.pyc and b/mlaas/__pycache__/__init__.cpython-310.pyc differ
diff --git a/mlaas/__pycache__/settings.cpython-310.pyc b/mlaas/__pycache__/settings.cpython-310.pyc
index 6f152b685e98d14ebd68ff2f44f9ab4e87e75e91..afa7918a7fba0b6903d56f75e547a0774f07bf2e 100644
Binary files a/mlaas/__pycache__/settings.cpython-310.pyc and b/mlaas/__pycache__/settings.cpython-310.pyc differ
diff --git a/mlaas/__pycache__/urls.cpython-310.pyc b/mlaas/__pycache__/urls.cpython-310.pyc
index ef9eab83ca1ae7e46bcef6ea27ed18e2fceef22f..0f1cf06cc17387db5fc3b6ce3beaff8da7b5ba97 100644
Binary files a/mlaas/__pycache__/urls.cpython-310.pyc and b/mlaas/__pycache__/urls.cpython-310.pyc differ
diff --git a/mlaas/__pycache__/wsgi.cpython-310.pyc b/mlaas/__pycache__/wsgi.cpython-310.pyc
index 70f4a29e89a284ebe7e900b1ac4c2c67ab445efb..8c010e9abb8de33c84380e6d40d9794de56d9741 100644
Binary files a/mlaas/__pycache__/wsgi.cpython-310.pyc and b/mlaas/__pycache__/wsgi.cpython-310.pyc differ
diff --git a/mlaas/middleware/__pycache__/__init__.cpython-310.pyc b/mlaas/middleware/__pycache__/__init__.cpython-310.pyc
index a80be63977ad5910e7c209186ab3d8ba89ac2f71..f4031dc3c609a0613bc4568e931319ff3b4a2e45 100644
Binary files a/mlaas/middleware/__pycache__/__init__.cpython-310.pyc and b/mlaas/middleware/__pycache__/__init__.cpython-310.pyc differ
diff --git a/mlaas/middleware/__pycache__/ai_engineer_auth.cpython-310.pyc b/mlaas/middleware/__pycache__/ai_engineer_auth.cpython-310.pyc
index 45823ddc49c942a525e50e2efe8ae2b7a7945231..37f38d54b71921538623fa6499e56c8d3a1f9bbe 100644
Binary files a/mlaas/middleware/__pycache__/ai_engineer_auth.cpython-310.pyc and b/mlaas/middleware/__pycache__/ai_engineer_auth.cpython-310.pyc differ
diff --git a/mlaas/middleware/__pycache__/autologout.cpython-310.pyc b/mlaas/middleware/__pycache__/autologout.cpython-310.pyc
index 65727ca390f13df45cba3317cf634599334efbf6..6cb715b609244b0367d6e3649578c5381fc69eae 100644
Binary files a/mlaas/middleware/__pycache__/autologout.cpython-310.pyc and b/mlaas/middleware/__pycache__/autologout.cpython-310.pyc differ
diff --git a/prediction_service/__pycache__/__init__.cpython-310.pyc b/prediction_service/__pycache__/__init__.cpython-310.pyc
index df9b7359cecf6be1e8ae77a54cb1634a6d22fb53..a1ebf17da9f5806b0882e96954f0200a228addc1 100644
Binary files a/prediction_service/__pycache__/__init__.cpython-310.pyc and b/prediction_service/__pycache__/__init__.cpython-310.pyc differ
diff --git a/prediction_service/__pycache__/admin.cpython-310.pyc b/prediction_service/__pycache__/admin.cpython-310.pyc
index f6551e6972cf72a0d6127a7f434be789c664307f..b9a646395e7aa62cd5a29a7c846cbb7af67e7ae4 100644
Binary files a/prediction_service/__pycache__/admin.cpython-310.pyc and b/prediction_service/__pycache__/admin.cpython-310.pyc differ
diff --git a/prediction_service/__pycache__/apps.cpython-310.pyc b/prediction_service/__pycache__/apps.cpython-310.pyc
index c1f4c9c51ebd3fad48dd469801b13007ce497049..288d7e8d4588c9ae7933d6992219cd8ac348aa4b 100644
Binary files a/prediction_service/__pycache__/apps.cpython-310.pyc and b/prediction_service/__pycache__/apps.cpython-310.pyc differ
diff --git a/prediction_service/__pycache__/forms.cpython-310.pyc b/prediction_service/__pycache__/forms.cpython-310.pyc
index 190616c0f68ba086659eab2f324dd4067026408c..dbd03b651113a52fbee094b609e82f433d4b3a67 100644
Binary files a/prediction_service/__pycache__/forms.cpython-310.pyc and b/prediction_service/__pycache__/forms.cpython-310.pyc differ
diff --git a/prediction_service/__pycache__/models.cpython-310.pyc b/prediction_service/__pycache__/models.cpython-310.pyc
index c4c54c42f7521f4556e19fd3de23d7c67fed8198..5a07257692a786e311b932998ab9bc0425a70f63 100644
Binary files a/prediction_service/__pycache__/models.cpython-310.pyc and b/prediction_service/__pycache__/models.cpython-310.pyc differ
diff --git a/prediction_service/__pycache__/urls.cpython-310.pyc b/prediction_service/__pycache__/urls.cpython-310.pyc
index 786e153a3fe304dd5d42013c2f692a1db2a6311c..eafe8ce1a61c30d842e1ca8709adf373ae9bb475 100644
Binary files a/prediction_service/__pycache__/urls.cpython-310.pyc and b/prediction_service/__pycache__/urls.cpython-310.pyc differ
diff --git a/prediction_service/__pycache__/views.cpython-310.pyc b/prediction_service/__pycache__/views.cpython-310.pyc
index 74b7ee27f814bc038df0e0cc011cab0046f32f13..a0e79055f79c7e743d0d1c2bc3fc4a97abd5f11f 100644
Binary files a/prediction_service/__pycache__/views.cpython-310.pyc and b/prediction_service/__pycache__/views.cpython-310.pyc differ
diff --git a/prediction_service/migrations/__pycache__/0001_initial.cpython-310.pyc b/prediction_service/migrations/__pycache__/0001_initial.cpython-310.pyc
index 42488715c8d7c57cdfb23c813c4dd69a63aa412b..1ca360e7eb072129f9c08021e025c0fe9501c6ab 100644
Binary files a/prediction_service/migrations/__pycache__/0001_initial.cpython-310.pyc and b/prediction_service/migrations/__pycache__/0001_initial.cpython-310.pyc differ
diff --git a/prediction_service/migrations/__pycache__/0002_remove_aiengineer_trained_models_and_more.cpython-310.pyc b/prediction_service/migrations/__pycache__/0002_remove_aiengineer_trained_models_and_more.cpython-310.pyc
index 5388c92c55b6da9493e5760368ebbb8bfa5ade48..3ecb775ae113a3b84febb490c6ac0b49edcfacc1 100644
Binary files a/prediction_service/migrations/__pycache__/0002_remove_aiengineer_trained_models_and_more.cpython-310.pyc and b/prediction_service/migrations/__pycache__/0002_remove_aiengineer_trained_models_and_more.cpython-310.pyc differ
diff --git a/prediction_service/migrations/__pycache__/0003_post.cpython-310.pyc b/prediction_service/migrations/__pycache__/0003_post.cpython-310.pyc
index 6560a8c4cea30243888184d2a9180735d231588c..0384cb34f0074cc2f0c9cca7c7ac8f6f1d21c49a 100644
Binary files a/prediction_service/migrations/__pycache__/0003_post.cpython-310.pyc and b/prediction_service/migrations/__pycache__/0003_post.cpython-310.pyc differ
diff --git a/prediction_service/migrations/__pycache__/0004_alter_userprofile_role.cpython-310.pyc b/prediction_service/migrations/__pycache__/0004_alter_userprofile_role.cpython-310.pyc
index 9ad9acaff84bfa620c29c81228e4e93e5956a34e..e4293edba4b8a8a30f1682ea9760208660fec9cc 100644
Binary files a/prediction_service/migrations/__pycache__/0004_alter_userprofile_role.cpython-310.pyc and b/prediction_service/migrations/__pycache__/0004_alter_userprofile_role.cpython-310.pyc differ
diff --git a/prediction_service/migrations/__pycache__/0005_alter_post_id_delete_financeteam.cpython-310.pyc b/prediction_service/migrations/__pycache__/0005_alter_post_id_delete_financeteam.cpython-310.pyc
index 7f01c9ebcfe6265154aac954fdc3bb71170fc670..317becd508dee2c77b8af76acdc9a9b3ca700eaa 100644
Binary files a/prediction_service/migrations/__pycache__/0005_alter_post_id_delete_financeteam.cpython-310.pyc and b/prediction_service/migrations/__pycache__/0005_alter_post_id_delete_financeteam.cpython-310.pyc differ
diff --git a/prediction_service/migrations/__pycache__/0006_mlmodel.cpython-310.pyc b/prediction_service/migrations/__pycache__/0006_mlmodel.cpython-310.pyc
index 2bae6b2cabdd138ee938f7c7e45d1e267363da43..16cf17cd0fd3ee7d53c5a8ab0d1db8df9b5b5ebc 100644
Binary files a/prediction_service/migrations/__pycache__/0006_mlmodel.cpython-310.pyc and b/prediction_service/migrations/__pycache__/0006_mlmodel.cpython-310.pyc differ
diff --git a/prediction_service/migrations/__pycache__/0007_post_mlmodel.cpython-310.pyc b/prediction_service/migrations/__pycache__/0007_post_mlmodel.cpython-310.pyc
index 41f47774ed95c98c1a95a5c297f47bb17d38ece4..b55630fedb4f26d1a79aa219754c78ce5461361d 100644
Binary files a/prediction_service/migrations/__pycache__/0007_post_mlmodel.cpython-310.pyc and b/prediction_service/migrations/__pycache__/0007_post_mlmodel.cpython-310.pyc differ
diff --git a/prediction_service/migrations/__pycache__/0008_post_slug.cpython-310.pyc b/prediction_service/migrations/__pycache__/0008_post_slug.cpython-310.pyc
index 6b0b8f53df8f3f268cd0781607619b3913310773..35c4970ef7f7415bb2a154e4aca9fa89818acb24 100644
Binary files a/prediction_service/migrations/__pycache__/0008_post_slug.cpython-310.pyc and b/prediction_service/migrations/__pycache__/0008_post_slug.cpython-310.pyc differ
diff --git a/prediction_service/migrations/__pycache__/0009_aiengineer_is_authorized.cpython-310.pyc b/prediction_service/migrations/__pycache__/0009_aiengineer_is_authorized.cpython-310.pyc
index 09036d11139adc6b6bf3a5609008d688f13d9f0c..ae362e8c6652fb1d0f16d1d1611f0f4f0b5ddd99 100644
Binary files a/prediction_service/migrations/__pycache__/0009_aiengineer_is_authorized.cpython-310.pyc and b/prediction_service/migrations/__pycache__/0009_aiengineer_is_authorized.cpython-310.pyc differ
diff --git a/prediction_service/migrations/__pycache__/0010_mlmodel_model_file.cpython-310.pyc b/prediction_service/migrations/__pycache__/0010_mlmodel_model_file.cpython-310.pyc
index 29367efe2bdd67ff3aa80678cb7d83873e844dc6..67f0b5fb6656ab17d889f0238e5e90ea7b6d3f45 100644
Binary files a/prediction_service/migrations/__pycache__/0010_mlmodel_model_file.cpython-310.pyc and b/prediction_service/migrations/__pycache__/0010_mlmodel_model_file.cpython-310.pyc differ
diff --git a/prediction_service/migrations/__pycache__/0011_alter_post_slug.cpython-310.pyc b/prediction_service/migrations/__pycache__/0011_alter_post_slug.cpython-310.pyc
index 546eb4f5089c37424b60adc0af4dae01f47abea2..74b157d91a009bbe5742f1ddccfafb0b32726477 100644
Binary files a/prediction_service/migrations/__pycache__/0011_alter_post_slug.cpython-310.pyc and b/prediction_service/migrations/__pycache__/0011_alter_post_slug.cpython-310.pyc differ
diff --git a/prediction_service/migrations/__pycache__/0012_alter_post_mlmodel.cpython-310.pyc b/prediction_service/migrations/__pycache__/0012_alter_post_mlmodel.cpython-310.pyc
index d5ee55750abcf3cb15e2d2628cb088db8c36cbf6..945ecadf9aab3f045f16ff6286c575e18a9149f2 100644
Binary files a/prediction_service/migrations/__pycache__/0012_alter_post_mlmodel.cpython-310.pyc and b/prediction_service/migrations/__pycache__/0012_alter_post_mlmodel.cpython-310.pyc differ
diff --git a/prediction_service/migrations/__pycache__/__init__.cpython-310.pyc b/prediction_service/migrations/__pycache__/__init__.cpython-310.pyc
index 135afc293d590cfddf6ca6b18c20878b6da9cd7d..0163eead76428e28007522bef161db0a1257ad71 100644
Binary files a/prediction_service/migrations/__pycache__/__init__.cpython-310.pyc and b/prediction_service/migrations/__pycache__/__init__.cpython-310.pyc differ
diff --git a/prediction_service/views.py b/prediction_service/views.py
index 0f87c30a215d40f18130b30ee06899ec8bd83d00..9b32fbe3fa5ab96b5e1740f62cea45a79f3951e5 100644
--- a/prediction_service/views.py
+++ b/prediction_service/views.py
@@ -214,7 +214,7 @@ def mlmodel(request, pk):
 
             # Get the ML model associated with the uploaded image
             try:
-                post = Post.objects.get(pk=pk)
+                post = Post.objects.get(mlmodel=pk)
                 mlmodel = post.mlmodel
             except Post.DoesNotExist:
                 return HttpResponse("Post not found.")