Update app.py

app.py

@@ -63,11 +63,12 @@ def show_results(input_image: Image.Image):
     perf_plot = "training_performance.png" if os.path.exists("training_performance.png") else None
     acc_plot = "final_test_accuracy.png" if os.path.exists("final_test_accuracy.png") else None
 
-    # Load final test accuracy number if available
-    test_acc_text = "Final Test Accuracy plot not found."
-    if acc_plot and os.path.exists("final_test_accuracy.png"):
-        # You can optionally parse accuracy from a saved file; here we just show a placeholder
-        test_acc_text = "See bar chart above for final test accuracy."
+    # Load final test accuracy number
+    test_acc_text = "Final test accuracy not available."
+    if os.path.exists("final_test_accuracy.txt"):
+        with open("final_test_accuracy.txt", "r") as f:
+            test_acc_value = f.read().strip()
+        test_acc_text = f"Final Test Accuracy: {test_acc_value}%"
 
     return preds, perf_plot, acc_plot, test_acc_text
 
@@ -84,7 +85,7 @@ interface = gr.Interface(
         gr.Textbox(label='Final Test Accuracy')
     ],
     title='CIFAR-10 Image Classification with DCLR Optimizer',
-    description='Upload an image to see predictions. Training/test plots show benchmark results on CIFAR-10.',
+    description='Upload an image to see predictions. Training/test plots and accuracy show benchmark results on CIFAR-10.',
     examples=example_images
 )