Update app.py

app.py

@@ -27,22 +27,55 @@ markdown_text = """
 - Input one budget line per time.
 - Accuracy of the model is ~88%.
 """
-markdown_table = """
-| Epoch | Training Loss | Validation Loss | Accuracy  |
-|-------|---------------|-----------------|-----------|
-| 1     | No log        | 2.095209        | 0.340764  |
-| 2     | No log        | 1.419945        | 0.662420  |
-| 3     | No log        | 0.683810        | 0.850318  |
-| 4     | No log        | 0.460408        | 0.872611  |
-| 5     | No log        | 0.422096        | 0.888535  |
+# HTML formatted table
+html_table = """
+<table>
+  <tr>
+    <th>Epoch</th>
+    <th>Training Loss</th>
+    <th>Validation Loss</th>
+    <th>Accuracy</th>
+  </tr>
+  <tr>
+    <td>1</td>
+    <td>No log</td>
+    <td>2.095209</td>
+    <td>0.340764</td>
+  </tr>
+  <tr>
+    <td>2</td>
+    <td>No log</td>
+    <td>1.419945</td>
+    <td>0.662420</td>
+  </tr>
+  <tr>
+    <td>3</td>
+    <td>No log</td>
+    <td>0.683810</td>
+    <td>0.850318</td>
+  </tr>
+  <tr>
+    <td>4</td>
+    <td>No log</td>
+    <td>0.460408</td>
+    <td>0.872611</td>
+  </tr>
+  <tr>
+    <td>5</td>
+    <td>No log</td>
+    <td>0.422096</td>
+    <td>0.888535</td>
+  </tr>
+</table>
 """
 
 iface = gr.Interface(
     fn=predict, 
     inputs=gr.components.Textbox(lines=1, placeholder="Enter Budget line here...", label="Budget Input"), 
-    outputs=[gr.components.Textbox(label="Classification Output"), gr.components.Article(markdown_table)],
+    outputs=gr.components.Label(label="Classification Output"), 
     title="COFOG Level 1 Classification",
     description=markdown_text,
+    article=html_table,
     allow_flagging="auto"  # Enables flagging
 )