Update app.py

app.py

@@ -259,13 +259,22 @@ thread.start()
 
 # Gradio Interface
 if prediction_plot_base64:
-    html_content = f'<img src="data:image/png;base64,{prediction_plot_base64}" alt="Prediction Plot">'
+    html_content = (
+    f'<img src="data:image/png;base64,{prediction_plot_base64}" '
+    'alt="Prediction Plot" '
+    'style="width: 100vw; height: 100vh; object-fit: contain;">'
+    )
 else:
     html_content = "Processing data..."
 
-# iface = gr.Interface(fn=None, inputs=None, outputs=gr.HTML(value=html_content))
-# iface.launch()
-with gr.Blocks() as demo:
+custom_css = """
+body, .gradio-container {
+    margin: 0;
+    padding: 0;
+}
+"""
+
+with gr.Blocks(css=custom_css) as demo:
     gr.HTML(value=html_content)
 
 demo.launch()