Update app.py

app.py

@@ -19,7 +19,7 @@ def predict(image,max_length=64,device='cpu'):
 
     entity_pair = get_entities(generated_caption)
 
-    knowlege_triplet = f"{entity_pair[0]}-{relation}->{entity_pair[1]}"
+    knowlege_triplet = f"'{entity_pair[0]}'---{relation}--->'{entity_pair[1]}'"
 
     return knowlege_triplet 
 
@@ -101,11 +101,12 @@ def get_relation(sent):
 
 
 
-input = gr.inputs.Image(label="Please upload a remote sensing image", type = 'pil', optional=True)
+input = gr.inputs.Image(label="Please upload an image", type = 'pil', optional=True)
 output = gr.outputs.Textbox(type="text",label="Captions")
 
 
-title = "Satellite Image Captioning"
+title = "Satellite Image Knowledge Extraction"
+description = "Provide an image taken from above, and receive back a corresponding head-relation-tail triplet that can be used to form a knowledge graph."
 
 interface = gr.Interface(
         fn=predict,
@@ -113,5 +114,6 @@ interface = gr.Interface(
         theme="grass",
         outputs=output,
         title=title,
+    
     )
 interface.launch(debug=True)