Update app.py

app.py

@@ -162,14 +162,14 @@ def infer(inputs, additional_inputs):
     return [gr.ScatterPlot(
             value=df_val,
             x="Prediction",
-            y="Target",
+            y="True Loss",
             title="Scatter",
-            tooltip=["Prediction", "Target"],
+            tooltip=["Prediction", "True Loss"],
             x_lim=[min(min(predictions), min(df_val[y_column]))-0.25, max(max(predictions), max(df_val[y_column]))+0.25],
             y_lim=[min(min(predictions), min(df_val[y_column]))-0.25, max(max(predictions), max(df_val[y_column]))+0.25]
         ), 
         gr.Image(Image.open('tmp.png')),
-        df_val[['Target', 'Prediction']], ]
+        df_val[['True Loss', 'Prediction']], ]
 
 def upload_csv(file):
     df = pd.read_csv(file.name,