Update app.py

app.py

@@ -15,11 +15,6 @@ def predict(image,max_length=50,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_relation(generated_caption)
-
-  #  entity_pair = get_entities(generated_caption)
-
-   # knowlege_triplet = f"'{entity_pair[0]}'---{relation}--->'{entity_pair[1]}'"
 
     return generated_caption