Update app.py

app.py

@@ -275,10 +275,27 @@ body, .gradio-container {
 }
 """
 
+# with gr.Blocks(css=custom_css) as demo:
+#     gr.HTML(value=html_content)
+
+
+# print("Data refreshed at:", datetime.datetime.now())
+
+# demo.launch()
+
 with gr.Blocks(css=custom_css) as demo:
-    gr.HTML(value=html_content)
+    html_output = gr.HTML()
 
+    def refresh_html():
+        if prediction_plot_base64:
+            return (
+                f'<img src="data:image/png;base64,{prediction_plot_base64}" '
+                'alt="Prediction Plot" '
+                'style="width: 100vw; height: 100vh; object-fit: contain;">'
+            )
+        else:
+            return "Processing data..."
 
-print("Data refreshed at:", datetime.datetime.now())
+    gr.Timer(refresh_html, every=10, every_units="seconds", show_progress=False, fn_kwargs={}, outputs=html_output)
 
-demo.launch()
+demo.launch()