diff --git a/gui/pages/Content/Models/MarketModels.js b/gui/pages/Content/Models/MarketModels.js
index 0ffe25ab5..09a717b67 100644
--- a/gui/pages/Content/Models/MarketModels.js
+++ b/gui/pages/Content/Models/MarketModels.js
@@ -43,35 +43,24 @@ export default function MarketModels(){
     useEffect(() => {
         loadingTextEffect('Loading Models', setLoadingText, 500);
         setIsLoading(true)
-        const iframe = document.getElementById('marketplace_models');
-
-        const handleIframeLoad = () => {
-            setTimeout(() => {
-                setIsLoading(false)
-            }, 2000);
-        };
-
-        if (iframe) {
-            iframe.onload = handleIframeLoad;
-        }
-
-        return () => {
-            if (iframe) {
-                iframe.onload = null;
-            }
-        };
+        setTimeout(() => {
+            setIsLoading(false)
+        }, 2500);
     }, []);
 
     return(
-        <div className="h_calc_sub_60 w_99vw">
-            {isLoading ? <iframe
-                id="marketplace_models"
-                src="https://models.superagi.com/marketplace"
-                width="100%"
-                height="100%"
-                frameBorder="0"
-                allowFullScreen
-            ></iframe>: <div className="horizontal_container_center h_75vh">
+        <div>
+            <div className={`${"h_calc_sub_60"} ${"w_99vw"} ${isLoading ? 'display_none' : 'display_block'}`}>
+                <iframe
+                    id="marketplace_models"
+                    src="http://localhost:3001/marketplace"
+                    width="100%"
+                    height="100%"
+                    frameBorder="0"
+                    allowFullScreen
+                ></iframe>
+            </div>
+            {isLoading && <div className="horizontal_container_center h_75vh">
                 <div className="signInInfo text_16 ff_sourceCode">{loadingText}</div>
             </div>}
         </div>
diff --git a/gui/pages/_app.css b/gui/pages/_app.css
index c949a6721..88e8d2dd3 100644
--- a/gui/pages/_app.css
+++ b/gui/pages/_app.css
@@ -1138,6 +1138,8 @@ p {
 
 .display_flex{display: inline-flex}
 .display_flex_container{display: flex}
+.display_none{display: none}
+.display_block{display: block}
 
 .align_center{align-items: center}
 .align_start{align-items: flex-start}