EAV123 commited on
Commit
d62c290
·
verified ·
1 Parent(s): 79ec034

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -192,11 +192,11 @@ elif page == "Data Upload":
192
  results_df.at[index, "Prediction"] = "Invalid data"
193
  continue
194
 
195
- # Extract only the required columns for prediction
196
  input_data = {
 
197
  'organism': row['organism'],
198
- 'antibiotic': row['antibiotic'],
199
- 'was_positive': row['was_positive']
200
  }
201
 
202
  # Get full prediction result
 
192
  results_df.at[index, "Prediction"] = "Invalid data"
193
  continue
194
 
195
+ # Extract only the required columns for prediction in specific order
196
  input_data = {
197
+ 'was_positive': row['was_positive'],
198
  'organism': row['organism'],
199
+ 'antibiotic': row['antibiotic']
 
200
  }
201
 
202
  # Get full prediction result