Update app.py

app.py

@@ -24,7 +24,7 @@ from torchvision import transforms
 _DEFAULT_IMAGE_TENSOR_NORMALIZATION_MEAN = [0.485, 0.456, 0.406]
 _DEFAULT_IMAGE_TENSOR_NORMALIZATION_STD = [0.229, 0.224, 0.225]
 
-def evaluate(img, id2entity, target_list, args):
+def evaluate(img):
     num_ent_id = len(id2entity)
     model = DistMult(args, num_ent_id, target_list, args.device)
     model.to(args.device)
@@ -158,7 +158,7 @@ if __name__=='__main__':
 
     species_model = gr.Interface(
         evaluate,
-        [gr.inputs.Image(shape=(200, 200)), id2entity, target_list, args],
+        gr.inputs.Image(shape=(200, 200)),
         outputs="label",
         title = 'Species Classification',
         description = 'Species Classification',