Update app.py

app.py

@@ -9,12 +9,13 @@ device='cpu'
 processor = AutoProcessor.from_pretrained("microsoft/git-base")
 model = AutoModelForCausalLM.from_pretrained("nkasmanoff/git-planet").to(device)
 
+nlp = spacy.load('en_core_web_sm')
 
 def predict(image,max_length=64,device='cpu'):
     pixel_values = processor(images=image, return_tensors="pt").to(device).pixel_values
     generated_ids = model.generate(pixel_values=pixel_values, max_length=max_length)
     generated_caption = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
-    relation = get_relations(generated_caption)
+    relation = get_relation(generated_caption)
 
     entity_pair = get_entities(generated_caption)
 
@@ -78,7 +79,6 @@ def get_entities(sent):
 
 
 def get_relation(sent):
-    nlp = spacy.load('en_core_web_sm')
 
     doc = nlp(sent)