WoodLB commited on
Commit
31e3ee7
·
1 Parent(s): ac157cd

finished app no frills

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -113,11 +113,12 @@ def train_and_inference(api_key, ontology_id, model_run_id):
113
  if data_split not in ['training', 'validation']: # we are skipping 'test' for now
114
  continue
115
 
116
- image_url = f"{entry['Labeled Data'].split('?')[0]}"
117
  label = entry['Label']['classifications'][0]['answer']['value']
118
 
119
  destination_folder = os.path.join(BASE_DIR, data_split, label)
120
  filename = os.path.basename(image_url)
 
121
 
122
  download_and_save_image(image_url, destination_folder, filename)
123
 
 
113
  if data_split not in ['training', 'validation']: # we are skipping 'test' for now
114
  continue
115
 
116
+ image_url = f"{entry['Labeled Data']}"
117
  label = entry['Label']['classifications'][0]['answer']['value']
118
 
119
  destination_folder = os.path.join(BASE_DIR, data_split, label)
120
  filename = os.path.basename(image_url)
121
+ st.write(filename)
122
 
123
  download_and_save_image(image_url, destination_folder, filename)
124