Update app.py

app.py

@@ -42,7 +42,7 @@ def aiornot0(image):
     AI: {px[0]}<br>
     Real: {px[1]}"""
     
-    return gr.update(html_out)
+    return gr.HTML.update(html_out)
 def aiornot1(image):    
     labels = ["Real", "AI"]
     mod=models[1]
@@ -80,6 +80,7 @@ with gr.Blocks() as app:
             
         with gr.Column():
             #outp0 = gr.Textbox(label=f'{models[0]}')
+            lab0 = gr.HTML(f"""Testing on Model {models[0]}""")
             outp0 = gr.HTML("""""")
             outp1 = gr.Textbox(label=f'{models[1]}')
             outp2 = gr.Textbox(label=f'{models[2]}')