WoodLB commited on
Commit
d21e9ab
·
1 Parent(s): 747cf34

finished app no frills

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -133,7 +133,7 @@ def train_and_inference(api_key, ontology_id, model_run_id):
133
 
134
  image_url = f"{entry['Labeled Data']}"
135
  label = entry['Label']['classifications'][0]['answer']['value']
136
- labeldict[f'{label.lower()}'] = label
137
 
138
  destination_folder = os.path.join(BASE_DIR, data_split, label)
139
  filename = os.path.basename(image_url)
 
133
 
134
  image_url = f"{entry['Labeled Data']}"
135
  label = entry['Label']['classifications'][0]['answer']['value']
136
+ labeldict[label] = entry['Label']['classifications'][0]['answer']['name']
137
 
138
  destination_folder = os.path.join(BASE_DIR, data_split, label)
139
  filename = os.path.basename(image_url)