snajmark commited on
Commit
68025fa
·
1 Parent(s): 81eb44a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -189,7 +189,8 @@ def predict_inverse(hardness_original_target, ys_original_target, metals_to_use,
189
  result[column] = round(result[column], 2)
190
 
191
  columns = list(result.columns)
192
- return result[columns[:-3]], result[columns[-3]], result[columns[0]], result[columns[1]], result[columns[-2]], result[columns[-1]]
 
193
 
194
 
195
  example_inputs = [820, 1800, ['Al', 'Fe']]
 
189
  result[column] = round(result[column], 2)
190
 
191
  columns = list(result.columns)
192
+ return (result[columns[2:-3]], result.at[columns[-3]], result.at[columns[0]],
193
+ result.at[columns[1]], result.at[columns[-2]], result.at[columns[-1]])
194
 
195
 
196
  example_inputs = [820, 1800, ['Al', 'Fe']]