Update app.py

app.py

@@ -29,6 +29,7 @@ id2entity = {v: k for k, v in entity2id.items()}
 datacsv = pd.read_csv('dataset_subtree.csv', low_memory=False)
 num_ent_id = len(entity2id)
 target_list = generate_target_list(datacsv, entity2id)  # Assuming this function is defined elsewhere
+overall_id_to_name = json.load(open('overall_id_to_name.json'))
 
 # Initialize your model here
 model = DistMult(num_ent_id, target_list, torch.device('cpu'))  # Update arguments as necessary
@@ -37,6 +38,7 @@ model.eval()
 # Define your evaluation function
 def evaluate(img):
     transform_steps = transforms.Compose([
+        transforms.ToPILImage(),
         transforms.Resize((448, 448)),
         transforms.ToTensor(),
         transforms.Normalize(_DEFAULT_IMAGE_TENSOR_NORMALIZATION_MEAN, _DEFAULT_IMAGE_TENSOR_NORMALIZATION_STD)
@@ -64,4 +66,4 @@ species_model = gr.Interface(
     description='Species Classification',
     article='Species Classification'
 )
-species_model.launch(share=True)
+species_model.launch(share=True)