Update app.py

app.py

@@ -26,11 +26,11 @@ def predict(text, url_params):
         mod_url = url_params.get('url')
         if mod_url != None:
             #print (mod_url)
-            out = gr.HTML.update(f"""<h1>Loading model from:<br>{mod_url} </h1>""")
+            out = gr.HTML.update(f"""<div><h1>Loading model from:<br>{mod_url} </h1></div>""")
         else:
-            out=gr.HTML.update(f"""<h1>Enter Model URL</h1>""") 
+            out=gr.HTML.update(f"""<div><h1>Enter Model URL</h1></div>""") 
     except Exception as e:
-        out=gr.HTML.update(f"""<h1>Enter Model URL</h1>""")
+        out=gr.HTML.update(f"""<div><h1>Enter Model URL</h1></div>""")
     
     return ["" + text + "", mod,out,mod_url]
    
@@ -41,18 +41,19 @@ def find_model(url):
         try:
             mod = gr.HTML(f'''
             <div>
-            <model-viewer src="{url}" camera-controls poster="https://cdn.glitch.global/fe6d4fc5-58dd-43de-a65d-7bd5b477e8e1/poster.webp?v=1696812691961" shadow-intensity="1">Could not Load</model-viewer>
+            <model-viewer src="{url}" camera-controls poster="https://cdn.glitch.global/fe6d4fc5-58dd-43de-a65d-7bd5b477e8e1/poster.webp?v=1696812691961" shadow-intensity="1"></model-viewer>
             </div>
             ''')
             
         except Exception as e:
-            out = gr.HTML(f'''<h1>Could not find Model</h1><br><h3>System Message: {e}''')
+            out = gr.HTML(f'''<div><h1>Could not find Model</h1><br><h3>System Message: {e}</div>''')
     else:
-        out = gr.HTML(f'''<h1>Enter Model URL</h1>''')
+        out = gr.HTML(f'''<div><h1>Enter Model URL</h1></div>''')
     return mod,out
 
     
 with gr.Blocks(css="style.css") as app:
+    markdown = gr.HTML("""""")
     with gr.Tab("GLB/GLTF"):
         with gr.Row():
             inp = gr.Textbox(label="URL")
@@ -66,7 +67,6 @@ with gr.Blocks(css="style.css") as app:
         </div>
         """)        
     '''
-    markdown = gr.HTML("""""")
 
     with gr.Row(visible=False):
         text_input=gr.Textbox()