modified: app.py

app.py

@@ -13,13 +13,12 @@ logits_dict = {}
 json_file = 'index.json'
 with open(json_file, 'r') as file:
     data = json.load(file)
-for item in data:
-    uuid = item['uuid']
-    text_description = item['text_description']
+for key, value in data.items():
+    text_description = value['text_description']
     inputs = tokenizer(text_description, return_tensors="pt", padding=True, truncation=True)
     outputs = model(**inputs, labels=inputs["input_ids"])
     logits = outputs.logits
-    logits_dict[uuid] = logits
+    logits_dict[key] = logits
 
 
 def search_index(query):