Update app.py

app.py

@@ -115,30 +115,30 @@ custom_css = """
         color: #333333;
         font-weight: bold;
         font-size: 24px;
-        margin-bottom: 10px.
+        margin-bottom: 10px;
     }
     .gradio-description {
         color: #666666;
         font-size: 16px;
-        margin-bottom: 20px.
+        margin-bottom: 20px;
     }
     .gradio-image {
-        border-radius: 10px.
+        border-radius: 10px;
     }
     .gradio-button {
         background-color: #007bff;
         color: #ffffff;
         border: none;
-        padding: 10px 20px.
-        border-radius: 5px.
-        cursor: pointer.
+        padding: 10px 20px;
+        border-radius: 5px;
+        cursor: pointer;
     }
     .gradio-button:hover {
-        background-color: #0056b3.
+        background-color: #0056b3;
     }
     .gradio-label {
-        color: #007bff.
-        font-weight: bold.
+        color: #007bff;
+        font-weight: bold;
     }
     </style>
 """
@@ -160,7 +160,7 @@ prediction_interface = gr.Interface(
 visualization_interface = gr.Interface(
     fn=display_visualizations,
     inputs=None,
-    outputs=[gr.Image(label=f"Visualization {i+1}") for i in range(len(visualization_images))],
+    outputs=[gr.Image(type="pil", label=f"Visualization {i+1}") for i in range(len(visualization_images))],
     title="Model Performance Visualizations",
     description=f"""
         Here are some visualizations that depict the performance of the model during training and testing.