michelecafagna26
/

clipcap-base-captioning-ft-hl-narratives

image-captioning

Inference Endpoints

Model card Files Files and versions Community

michelecafagna26 commited on Jul 24, 2023

Commit

3d59069

·

1 Parent(s): 9ab61b3

Update README.md

Files changed (1) hide show

README.md +0 -1

README.md CHANGED Viewed

@@ -54,7 +54,6 @@ git clone https://huggingface.co/michelecafagna26/clipcap-base-captioning-ft-hl-
 ```python
 from clipcap import ClipCaptionModel
-import torch
 from transformers import (
     GPT2Tokenizer,
     GPT2LMHeadModel,

 ```python
 from clipcap import ClipCaptionModel
 from transformers import (
     GPT2Tokenizer,
     GPT2LMHeadModel,