Spaces:

zeddotes
/

computer-thoughts

Sleeping

zeddotes commited on Feb 22

Commit

3454c90

1 Parent(s): 5c85d5d

updated app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,8 +4,8 @@ from PIL import Image
 import torch
 # Example with BLIP (replace with your fine-tuned model)
-processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
-model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
 def caption_image(image):
     if image is None:

 import torch
 # Example with BLIP (replace with your fine-tuned model)
+processor = BlipProcessor.from_pretrained("zeddotes/blip-computer-thoughts")
+model = BlipForConditionalGeneration.from_pretrained("zeddotes/blip-computer-thoughts")
 def caption_image(image):
     if image is None: