Update app.py

app.py

@@ -1,6 +1,6 @@
 import torch
 from ram import get_transform, inference_ram, inference_tag2text
-from ram.models import ram, tag2text_caption
+from ram.models import ram, tag2text
 
 ram_checkpoint = "./ram_swin_large_14m.pth"
 tag2text_checkpoint = "./tag2text_swin_14m.pth"
@@ -44,7 +44,7 @@ if __name__ == "__main__":
     # get transform and load models
     transform = get_transform(image_size=image_size)
     ram_model = ram(pretrained=ram_checkpoint, image_size=image_size, vit='swin_l').eval().to(device)
-    tag2text_model = tag2text_caption(
+    tag2text_model = tag2text(
         pretrained=tag2text_checkpoint, image_size=image_size, vit='swin_b').eval().to(device)
 
     # build GUI