Update app.py

app.py

@@ -13,8 +13,13 @@ 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]
-    
-    return generated_caption 
+    relation = get_relations(generated_caption)
+
+    entity_pair = get_entities(generated_caption)
+
+    knowlege_triplet = f"{entity_pair[0]}-{relation}->{entity_pair[1]}"
+
+    return knowlege_triplet 
 
 
 input = gr.inputs.Image(label="Please upload a remote sensing image", type = 'pil', optional=True)