Update app.py

app.py

@@ -189,8 +189,8 @@ def predict_inverse(hardness_original_target, ys_original_target, metals_to_use,
             result[column] = round(result[column], 2)
 
     columns = list(result.columns)
-    return (result[columns[2:-3]], result.at[columns[-3]], result.at[columns[0]],
-            result.at[columns[1]], result.at[columns[-2]], result.at[columns[-1]])
+    return (result[columns[2:-3]], result.at[0, columns[-3]], result.at[0, columns[0]],
+            result.at[0, columns[1]], result.at[0, columns[-2]], result.at[0, columns[-1]])
 
 
 example_inputs = [820, 1800, ['Al', 'Fe']]