Spaces:

SaadNadeem07
/

BLIP_2-VLM-Finetuning

Running

Saadi07 commited on May 1

Commit

eb89eec

1 Parent(s): ebc7baf

Model Deployed

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ from transformers import Blip2ForConditionalGeneration, AutoProcessor
 # Load your fine-tuned model and processor from local directories
 processor = AutoProcessor.from_pretrained("./processor")
-model = Blip2ForConditionalGeneration.from_pretrained("./model", device_map="auto", torch_dtype=torch.float16)
 # Inference function
 def generate_caption(image: Image.Image) -> str:

 # Load your fine-tuned model and processor from local directories
 processor = AutoProcessor.from_pretrained("./processor")
+model = Blip2ForConditionalGeneration.from_pretrained("./model")
 # Inference function
 def generate_caption(image: Image.Image) -> str: